Update app.py

app.py

@@ -60,56 +60,7 @@ setup_eval_logging()
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device="cpu")
 PIXABAY_API_KEY = "33492762-a28a596ec4f286f84cd328b17"
 
-# CSS 스타일 정의
-custom_css = """
-.gradio-container {
-    background: linear-gradient(45deg, #1a1a1a, #2a2a2a);
-    border-radius: 15px;
-    box-shadow: 0 8px 32px rgba(0,0,0,0.3);
-}
 
-.input-container, .output-container {
-    background: rgba(255,255,255,0.1);
-    backdrop-filter: blur(10px);
-    border-radius: 10px;
-    padding: 20px;
-    transform-style: preserve-3d;
-    transition: transform 0.3s ease;
-}
-
-.input-container:hover {
-    transform: translateZ(20px);
-}
-
-.gallery-item {
-    transition: transform 0.3s ease;
-    border-radius: 8px;
-    overflow: hidden;
-}
-
-.gallery-item:hover {
-    transform: scale(1.05);
-    box-shadow: 0 4px 15px rgba(0,0,0,0.2);
-}
-
-.tabs {
-    background: rgba(255,255,255,0.05);
-    border-radius: 10px;
-    padding: 10px;
-}
-
-button {
-    background: linear-gradient(45deg, #4a90e2, #357abd);
-    border: none;
-    border-radius: 5px;
-    transition: all 0.3s ease;
-}
-
-button:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 4px 15px rgba(74,144,226,0.3);
-}
-"""
 
 def cleanup_temp_files():
     temp_dir = tempfile.gettempdir()
@@ -266,11 +217,91 @@ def text_to_audio(prompt: str, negative_prompt: str, seed: int, num_steps: int,
     torchaudio.save(audio_save_path, audio, seq_cfg.sampling_rate)
     return audio_save_path
 
-# 인터페이스 정의
+
+
+# CSS 스타일 수정
+custom_css = """
+.gradio-container {
+    background: linear-gradient(45deg, #1a1a1a, #2a2a2a);
+    border-radius: 15px;
+    box-shadow: 0 8px 32px rgba(0,0,0,0.3);
+    color: #e0e0e0;
+}
+
+.input-container, .output-container {
+    background: rgba(40, 40, 40, 0.95);
+    backdrop-filter: blur(10px);
+    border-radius: 10px;
+    padding: 20px;
+    transform-style: preserve-3d;
+    transition: transform 0.3s ease;
+    border: 1px solid rgba(255, 255, 255, 0.1);
+}
+
+.input-container:hover {
+    transform: translateZ(20px);
+    box-shadow: 0 8px 32px rgba(0,0,0,0.5);
+}
+
+.gallery-item {
+    transition: transform 0.3s ease;
+    border-radius: 8px;
+    overflow: hidden;
+    background: #2a2a2a;
+}
+
+.gallery-item:hover {
+    transform: scale(1.05);
+    box-shadow: 0 4px 15px rgba(0,0,0,0.4);
+}
+
+.tabs {
+    background: rgba(30, 30, 30, 0.95);
+    border-radius: 10px;
+    padding: 10px;
+    border: 1px solid rgba(255, 255, 255, 0.05);
+}
+
+button {
+    background: linear-gradient(45deg, #2196F3, #1976D2);
+    border: none;
+    border-radius: 5px;
+    transition: all 0.3s ease;
+    color: white;
+}
+
+button:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 4px 15px rgba(33,150,243,0.3);
+}
+
+/* 텍스트 입력 필드 스타일 */
+textarea, input[type="text"], input[type="number"] {
+    background: rgba(30, 30, 30, 0.95) !important;
+    border: 1px solid rgba(255, 255, 255, 0.1) !important;
+    color: #e0e0e0 !important;
+    border-radius: 5px !important;
+}
+
+/* 레이블 스타일 */
+label {
+    color: #e0e0e0 !important;
+}
+
+/* 갤러리 그리드 스타일 */
+.gallery {
+    background: rgba(30, 30, 30, 0.95);
+    padding: 15px;
+    border-radius: 10px;
+    border: 1px solid rgba(255, 255, 255, 0.05);
+}
+"""
+
+# 인터페이스 정의 수정 (영문으로 변경)
 video_search_tab = gr.Interface(
     fn=search_videos,
-    inputs=gr.Textbox(label="검색어 입력"),
-    outputs=gr.Gallery(label="검색 결과", columns=4, rows=20),
+    inputs=gr.Textbox(label="Search Query(한글지원)"),
+    outputs=gr.Gallery(label="Search Results", columns=4, rows=20),
     css=custom_css,
     api_name=False
 )
@@ -278,36 +309,37 @@ video_search_tab = gr.Interface(
 video_to_audio_tab = gr.Interface(
     fn=video_to_audio,
     inputs=[
-        gr.Video(label="비디오"),
-        gr.Textbox(label="프롬프트"),
-        gr.Textbox(label="네거티브 프롬프트", value="music"),
-        gr.Number(label="시드", value=0),
-        gr.Number(label="스텝 수", value=25),
-        gr.Number(label="가이드 강도", value=4.5),
-        gr.Number(label="길이(초)", value=8),
+        gr.Video(label="Input Video"),
+        gr.Textbox(label="Prompt(한글지원)"),
+        gr.Textbox(label="Negative Prompt", value="music"),
+        gr.Number(label="Seed", value=0),
+        gr.Number(label="Steps", value=25),
+        gr.Number(label="Guidance Scale", value=4.5),
+        gr.Number(label="Duration (sec)", value=8),
     ],
-    outputs="playable_video",
+    outputs=gr.Video(label="Generated Result"),
     css=custom_css
 )
 
 text_to_audio_tab = gr.Interface(
     fn=text_to_audio,
     inputs=[
-        gr.Textbox(label="프롬프트"),
-        gr.Textbox(label="네거티브 프롬프트"),
-        gr.Number(label="시드", value=0),
-        gr.Number(label="스텝 수", value=25),
-        gr.Number(label="가이드 강도", value=4.5),
-        gr.Number(label="길이(초)", value=8),
+        gr.Textbox(label="Prompt(한글지원)"),
+        gr.Textbox(label="Negative Prompt"),
+        gr.Number(label="Seed", value=0),
+        gr.Number(label="Steps", value=25),
+        gr.Number(label="Guidance Scale", value=4.5),
+        gr.Number(label="Duration (sec)", value=8),
     ],
-    outputs="audio",
+    outputs=gr.Audio(label="Generated Audio"),
     css=custom_css
 )
 
-# 메인 실행
+# 메인 실행 부분 수정
 if __name__ == "__main__":
     gr.TabbedInterface(
         [video_search_tab, video_to_audio_tab, text_to_audio_tab],
-        ["비디오 검색", "비디오-오디오 변환", "텍스트-오디오 변환"],
+        ["Video Search", "Video-to-Audio", "Text-to-Audio"],
         css=custom_css
-    ).launch(allowed_paths=[output_dir])
+    ).launch(allowed_paths=[output_dir])
+