Update app.py

app.py

@@ -66,7 +66,7 @@ print("PATH:", os.environ['PATH'])
 print("LD_LIBRARY_PATH:", os.environ['LD_LIBRARY_PATH'])
 
 from stf_utils import STFPipeline
-from flux_schnell import create_flux_tab
+
 
 
 # audio_path="assets/examples/driving/test_aud.mp3"
@@ -186,7 +186,7 @@ output_image_paste_back = gr.Image(type="numpy")
 output_video = gr.Video()
 output_video_concat = gr.Video()
 
-# video_input = gr.Video()
+video_input = gr.Video()
 driving_video_path=gr.Video()
 
 
@@ -213,7 +213,7 @@ 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")
+                    #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()
@@ -240,7 +240,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                         cache_examples=False,
                     )
                 with gr.Accordion(open=True, label="Driving Video"):
-                    video_input = gr.Video()
+                    #video_input = gr.Video()
                     gr.Examples(
                         examples=[
                             [osp.join(example_video_dir, "d0.mp4")],
@@ -323,10 +323,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             video_gen_button.click(
                 fn=gpu_wrapped_stf_pipeline_execute,
                 inputs=[
-                    #output_audio
-                    audio_path_component
+                    output_audio
+                    #audio_path_component
+                ],
+                outputs=[
+                    video_input
+                    #driving_video_path
                 ],
-                outputs=[driving_video_path],
                 show_progress=True
             )
 
@@ -343,9 +346,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,