Add a status bar to the Gradio UI

app.py

@@ -172,15 +172,17 @@ def get_gemini_response(model_name: str, encoded_images: List[Dict], prompt: str
     except Exception as e:
         return f"Error with {model_name}: {str(e)}"
 
-def process_input(id_product_money: str, prompt: str) -> Tuple[List[str], str, str]:
+def process_input(id_product_money: str, prompt: str, progress=gr.Progress()) -> Tuple[List[str], str, str, str]:
     """Main processing function."""
     try:
-        print("Getting marketplace and main image...")
+        status_msg = "Getting product data from database..."
+        progress(0, desc=status_msg)
         marketplace, main_image = get_marketplace_and_main_image(id_product_money)
         print(f"Marketplace: {marketplace}")
         print(f"Main image: {main_image}")
         
-        print("\nGetting additional images...")
+        status_msg = "Fetching additional product images..."
+        progress(0.2, desc=status_msg)
         additional_images = get_additional_images(id_product_money, marketplace)
         print(f"Additional images: {additional_images}")
         
@@ -193,23 +195,31 @@ def process_input(id_product_money: str, prompt: str) -> Tuple[List[str], str, s
                 all_image_urls.append(url)
         print(f"\nAll image URLs: {all_image_urls}")
         
-        print("\nDownloading and encoding images...")
+        status_msg = "Downloading and processing images..."
+        progress(0.4, desc=status_msg)
         encoded_images = download_and_encode_images(all_image_urls, marketplace)
         print(f"Number of encoded images: {len(encoded_images)}")
         
         if not encoded_images:
             raise ValueError("No images could be downloaded")
         
-        print("\nGetting Gemini responses...")
-        # Get responses from both models
+        status_msg = "Getting response from Gemini 1.5 Flash..."
+        progress(0.6, desc=status_msg)
         gemini_1_5_response = get_gemini_response("gemini-1.5-flash", encoded_images, prompt)
+        
+        status_msg = "Getting response from Gemini 2.0 Flash Exp..."
+        progress(0.8, desc=status_msg)
         gemini_2_0_response = get_gemini_response("gemini-2.0-flash-exp", encoded_images, prompt)
         
-        return all_image_urls, gemini_1_5_response, gemini_2_0_response
+        status_msg = "Analysis complete!"
+        progress(1.0, desc=status_msg)
+        return all_image_urls, gemini_1_5_response, gemini_2_0_response, status_msg
     
     except Exception as e:
         print(f"\nError in process_input: {str(e)}")
-        return [], f"Error: {str(e)}", f"Error: {str(e)}"
+        status_msg = f"Error: {str(e)}"
+        progress(1.0, desc=status_msg)
+        return [], f"Error: {str(e)}", f"Error: {str(e)}", status_msg
 
 # Create Gradio interface
 with gr.Blocks() as demo:
@@ -221,6 +231,9 @@ with gr.Blocks() as demo:
     
     submit_btn = gr.Button("Analyze")
     
+    # Status indicator
+    status = gr.Textbox(label="Status", value="Waiting for input...", interactive=False)
+    
     with gr.Row():
         image_gallery = gr.Gallery(label="Product Images", show_label=True)
     
@@ -235,7 +248,8 @@ with gr.Blocks() as demo:
     submit_btn.click(
         fn=process_input,
         inputs=[id_input, prompt_input],
-        outputs=[image_gallery, gemini_1_5_output, gemini_2_0_output]
+        outputs=[image_gallery, gemini_1_5_output, gemini_2_0_output, status],
+        show_progress="full"
     )
 
 if __name__ == "__main__":