Update app.py

app.py

@@ -55,6 +55,25 @@ def identify_and_save_blob(blob_path):
         raise ValueError(f"An error occurred while processing the file: {e}")
 
 
+def process_file_upload(file_path):
+    """Process uploaded file and return the file path and image if applicable"""
+    if isinstance(file_path, str):
+        if file_path.endswith(tuple([i for i, f in image_extensions.items()])):
+            return file_path, Image.open(file_path)
+        elif file_path.endswith(video_extensions):
+            return file_path, None
+        else:
+            try:
+                media_path, media_type = identify_and_save_blob(file_path)
+                if media_type == "image":
+                    return media_path, Image.open(media_path)
+                return media_path, None
+            except Exception as e:
+                print(e)
+                raise ValueError("Unsupported media type. Please upload an image or video.")
+    return None, None
+
+
 @spaces.GPU
 def qwen_inference(media_input, text_input=None):
     if isinstance(media_input, str):  # If it's a filepath
@@ -130,13 +149,20 @@ 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"
                 )
+                preview_image = gr.Image(label="Preview", visible=True)
                 text_input = gr.Textbox(label="Question")
                 submit_btn = gr.Button(value="Submit")
             with gr.Column():
                 output_text = gr.Textbox(label="Output Text")
 
+        input_media.change(
+            fn=process_file_upload,
+            inputs=[input_media],
+            outputs=[input_media, preview_image]
+        )
+        
         submit_btn.click(
             qwen_inference, [input_media, text_input], [output_text]
         )