Update app.py

app.py

@@ -22,8 +22,10 @@ from elevenlabs_utils import ElevenLabsPipeline
 from setup_environment import initialize_environment
 from src.utils.video import extract_audio
 #from flux_dev import create_flux_tab
-from flux_schnell import create_flux_tab
+#from flux_schnell import create_flux_tab
 # from diffusers import FluxPipeline
+import os
+import sys
 
 # import gdown
 # folder_url = f"https://drive.google.com/drive/folders/1UtKgzKjFAOmZkhNK-OYT0caJ_w2XAnib"
@@ -35,19 +37,27 @@ from flux_schnell import create_flux_tab
     
 initialize_environment()
 
-import sys
-sys.path.append('/home/user/.local/lib/python3.10/site-packages')
-sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_alternative/src/stf_alternative')
-sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_tools/src/stf_tools')
-sys.path.append('/home/user/app/')
-sys.path.append('/home/user/app/stf/')
-sys.path.append('/home/user/app/stf/stf_alternative/')
-sys.path.append('/home/user/app/stf/stf_alternative/src/stf_alternative')
-sys.path.append('/home/user/app/stf/stf_tools')
-sys.path.append('/home/user/app/stf/stf_tools/src/stf_tools')
+
+sys.path.extend([
+    '/home/user/.local/lib/python3.10/site-packages',
+    '/home/user/app/',
+    '/home/user/app/stf/',
+    '/home/user/app/stf/stf_alternative/',
+    '/home/user/app/stf/stf_tools'
+])
+
+# sys.path.append('/home/user/.local/lib/python3.10/site-packages')
+# sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_alternative/src/stf_alternative')
+# sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_tools/src/stf_tools')
+# sys.path.append('/home/user/app/')
+# sys.path.append('/home/user/app/stf/')
+# sys.path.append('/home/user/app/stf/stf_alternative/')
+# sys.path.append('/home/user/app/stf/stf_alternative/src/stf_alternative')
+# sys.path.append('/home/user/app/stf/stf_tools')
+# sys.path.append('/home/user/app/stf/stf_tools/src/stf_tools')
+
 
 
-import os
 # CUDA 경로를 환경 변수로 설정
 os.environ['PATH'] = '/usr/local/cuda/bin:' + os.environ.get('PATH', '')
 os.environ['LD_LIBRARY_PATH'] = '/usr/local/cuda/lib64:' + os.environ.get('LD_LIBRARY_PATH', '')
@@ -116,7 +126,7 @@ gradio_pipeline = GradioPipeline(
 
 # 추가 정의
 elevenlabs_pipeline = ElevenLabsPipeline()
-#stf_pipeline = STFPipeline()
+stf_pipeline = STFPipeline()
 
 
 @spaces.GPU() #duration=240)
@@ -202,12 +212,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             
             
             gr.Markdown("### 2. Audio to Driving-Video")
-            # with gr.Row():
-            #         audio_path_component = gr.Textbox(label="Input", value="assets/examples/driving/test_aud.mp3")
-            #         video_gen_button = gr.Button("Audio to Video generation", variant="primary")
-            # with gr.Row():
-            #         #a2v_output = gr.Video()
-            #         driving_video_path.render()
+            with gr.Row():
+                    audio_path_component = gr.Textbox(label="Input", value="assets/examples/driving/test_aud.mp3")
+                    video_gen_button = gr.Button("Audio to Video generation", variant="primary")
+            with gr.Row():
+                    #a2v_output = gr.Video()
+                    driving_video_path.render()
 
                 
             gr.Markdown("### 3. Image to Talking Video")
@@ -310,15 +320,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 show_progress=True
             )
 
-            # video_gen_button.click(
-            #     fn=gpu_wrapped_stf_pipeline_execute,
-            #     inputs=[
-            #         #output_audio
-            #         audio_path_component
-            #     ],
-            #     outputs=[driving_video_path],
-            #     show_progress=True
-            # )
+            video_gen_button.click(
+                fn=gpu_wrapped_stf_pipeline_execute,
+                inputs=[
+                    #output_audio
+                    audio_path_component
+                ],
+                outputs=[driving_video_path],
+                show_progress=True
+            )
 
 
             
@@ -333,9 +343,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 outputs=video_input
             )
         
-        # 세 번째 탭: Flux 개발용 탭
-        with gr.Tab("FLUX Image"):
-            flux_demo = create_flux_tab(image_input)  # Flux 개발용 탭 생성
+        # # 세 번째 탭: Flux 개발용 탭
+        # with gr.Tab("FLUX Image"):
+        #     flux_demo = create_flux_tab(image_input)  # Flux 개발용 탭 생성
 
 demo.launch(
     server_port=args.server_port,