Update interface/app.py

interface/app.py

@@ -144,23 +144,32 @@ def create_interface():
         #     inputs=[processing_status],
         #     outputs=[results_box, selected_images, confidence_scores, processing_status]
         # )
+
+
+        progress_message = gr.Markdown("Ready to analyze...")
+
+        
         # Update the click handler
+
         submit_btn.click(
-            # First clear/reset outputs
+            # First update UI to show processing state
             fn=lambda: (
-                "Processing...",  # for results_box
-                None,            # for selected_images
-                None,            # for confidence_scores
-                {"status": "processing"}  # for processing_status
+                gr.update(visible=True, value="⏳ Processing your request..."),  # progress message
+                gr.update(visible=False),  # results
+                gr.update(visible=False),  # images
+                gr.update(visible=False),  # scores
+                gr.update(visible=True, value={"status": "started", "step": "initializing"})  # status
             ),
             inputs=None,
             outputs=[
+                progress_message,
                 results_box,
                 selected_images,
                 confidence_scores,
                 processing_status
             ]
-        ).then(  # Then process inputs
+        ).then(
+            # Process inputs
             fn=handler.process_inputs,
             inputs=[
                 query,
@@ -174,8 +183,52 @@ def create_interface():
                 selected_images,
                 confidence_scores,
                 processing_status
-            ]
+            ],
+            show_progress="full"  # Show gradio's built-in progress
+        ).then(
+            # Update UI after processing
+            fn=lambda: gr.update(visible=True, value="✅ Analysis complete!"),
+            inputs=None,
+            outputs=progress_message
+        )
+
+        # Add error handling feedback
+        progress_message.change(
+            fn=lambda x: gr.update(visible=True) if "Error" in str(x) else gr.update(visible=False),
+            inputs=[progress_message],
+            outputs=[progress_message]
         )
+        # submit_btn.click(
+        #     # First clear/reset outputs
+        #     fn=lambda: (
+        #         "Processing...",  # for results_box
+        #         None,            # for selected_images
+        #         None,            # for confidence_scores
+        #         {"status": "processing"}  # for processing_status
+        #     ),
+        #     inputs=None,
+        #     outputs=[
+        #         results_box,
+        #         selected_images,
+        #         confidence_scores,
+        #         processing_status
+        #     ]
+        # ).then(  # Then process inputs
+        #     fn=handler.process_inputs,
+        #     inputs=[
+        #         query,
+        #         constraints,
+        #         images,
+        #         top_k,
+        #         report_format
+        #     ],
+        #     outputs=[
+        #         results_box,
+        #         selected_images,
+        #         confidence_scores,
+        #         processing_status
+        #     ]
+        # )
 
     return app