Update flux_dev.py

flux_dev.py

@@ -44,7 +44,8 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
         yield img, seed
 
 
-def create_flux_tab():
+#def create_flux_tab():
+def create_flux_tab(image_input):
     examples = [
         "a tiny astronaut hatching from an egg on the moon",
         "a cat holding a sign that says hello world",
@@ -72,6 +73,9 @@ def create_flux_tab():
                 run_button = gr.Button("Run", scale=0)
             
             result = gr.Image(label="Result", show_label=False)
+
+            with gr.Row():
+                use_in_text2lipsync_button = gr.Button("Use this image in [Txt2LipSync] Tab")  # 새로운 버튼 추가
             
             with gr.Accordion("Advanced Settings", open=False):
                 seed = gr.Slider(
@@ -129,5 +133,11 @@ def create_flux_tab():
             inputs=[prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
             outputs=[result, seed]
         )
+        # 새로운 버튼 클릭 이벤트 정의
+        use_in_text2lipsync_button.click(
+            fn=lambda img: img,  # 간단한 람다 함수를 사용하여 이미지를 그대로 전달
+            inputs=[result],  # 생성된 이미지를 입력으로 사용
+            outputs=[image_input]  # Text to LipSync 탭의 image_input을 업데이트
+        )
     
     return flux_demo