Update app.py

app.py

@@ -118,10 +118,15 @@ with gr.Blocks() as demo:
         persp_input = gr.File(label="Upload Perspective Image", file_types=[".jpg",".png",".jpeg"])
         json_input = gr.File(label="Upload mockup.json", file_types=[".json"])
     with gr.Row():
-        gr.Image(type="filepath", label="Flat Image Preview").style(height=200)
-        gr.Image(type="filepath", label="Perspective Image Preview").style(height=200)
+        flat_preview = gr.Image(type="filepath", label="Flat Image Preview", height=200)
+        persp_preview = gr.Image(type="filepath", label="Perspective Image Preview", height=200)
     output_gallery = gr.Gallery(label="Perspective ROI Results", show_label=True, columns=2, height=600)
     run_btn = gr.Button("Run Homography")
+
+    # Link previews
+    flat_input.change(lambda f: f.name if f else None, inputs=flat_input, outputs=flat_preview)
+    persp_input.change(lambda f: f.name if f else None, inputs=persp_input, outputs=persp_preview)
+
     run_btn.click(homography_demo, inputs=[flat_input, persp_input, json_input], outputs=output_gallery)
 
 demo.launch()