Update app.py

app.py

@@ -443,7 +443,6 @@ def load_preview(url: str):
     except Exception:
         return None, None, "Preview failed"
 
-
 with gr.Blocks(title="Flux", css=css) as demo:
     with gr.Row():
         with gr.Column(scale=1):
@@ -452,4 +451,17 @@ with gr.Blocks(title="Flux", css=css) as demo:
             with gr.Accordion("Mistral API Key (optional)", open=False):
                 api_key = gr.Textbox(label="API Key", type="password", max_lines=1)
             submit = gr.Button("Submit")
-            preview_image = gr.Image(label="Preview image", type="pil", visible
+            preview_image = gr.Image(label="Preview", type="pil", elem_classes="preview_media")
+            preview_video = gr.Video(label="Preview", elem_classes="preview_media")
+
+        with gr.Column(scale=2):
+            final_text = gr.Markdown(value="")
+
+    # Ensure preview outputs get None when not applicable so Gradio hides them
+    url_input.change(fn=load_preview, inputs=[url_input], outputs=[preview_image, preview_video, gr.Textbox(visible=False)])
+    submit.click(fn=generate_final_text, inputs=[url_input, custom_prompt, api_key], outputs=[final_text])
+    demo.queue()
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
+