Update app.py

app.py

@@ -115,6 +115,7 @@ def qwen_inference(media_input, text_input=None):
         buffer += new_text
         yield buffer
 
+
 css = """
   #output {
     height: 500px; 
@@ -130,15 +131,28 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             with gr.Column():
                 input_media = gr.File(
-                    label="Upload Image or Video", type="filepath" 
+                    label="Upload Image or Video", type="filepath"
                 )
+                image_viewer = gr.Image(label="Uploaded Image", interactive=False, visible=False)
                 text_input = gr.Textbox(label="Question")
                 submit_btn = gr.Button(value="Submit")
             with gr.Column():
                 output_text = gr.Textbox(label="Output Text")
 
+        # Display the uploaded image
+        def display_uploaded_image(media_input):
+            if media_input.endswith(tuple([i for i, f in image_extensions.items()])):
+                return media_input, True  # Show image
+            return None, False  # Hide image
+        
+        input_media.change(
+            display_uploaded_image,
+            inputs=input_media,
+            outputs=[image_viewer, image_viewer.visible]
+        )
+
         submit_btn.click(
             qwen_inference, [input_media, text_input], [output_text]
         )
 
-demo.launch(debug=True)
+demo.launch(debug=True)