Update app.py

app.py

@@ -1,5 +1,6 @@
-# Add this function before the Gradio interface definition (around line 680)
+return combined_content, output_files, "✅ Analysis completed successfully!"
 
+# Wrapper function for Gradio interface
 def process_dashboard(api_key, pdf_files, language_name, goal_description=None, num_sections=4, model_name=DEFAULT_MODEL, custom_model=None):
     """Process dashboard PDFs and generate reports (wrapper function for Gradio interface)."""
     # Start a thread to update progress
@@ -51,4 +52,109 @@ def process_dashboard(api_key, pdf_files, language_name, goal_description=None,
         print(error_message)
         progress_tracker.update(100, error_message)
         progress_tracker.end_processing()
-        return None, None, error_message
+        return None, None, error_message
+
+# Gradio Interface Functions
+def toggle_custom_model(choice):
+    """Toggle visibility of custom model input based on selection."""
+    return {"visible": choice == "custom"}
+
+def refresh_models(api_key):
+    """Refresh the list of available models based on the API key."""
+    if not api_key:
+        return gr.Dropdown(choices=["meta-llama/llama-4-scout:free", "custom"] + OPENROUTER_MODELS, value="meta-llama/llama-4-scout:free")
+    
+    try:
+        available_models = get_available_models(api_key)
+        return gr.Dropdown(choices=["meta-llama/llama-4-scout:free", "custom"] + available_models, value="meta-llama/llama-4-scout:free")
+    except Exception as e:
+        print(f"Error refreshing models: {str(e)}")
+        return gr.Dropdown(choices=["meta-llama/llama-4-scout:free", "custom"] + OPENROUTER_MODELS, value="meta-llama/llama-4-scout:free")
+
+# Define the Gradio interface
+with gr.Blocks(title="Dashboard Narrator - Powered by OpenRouter.ai", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("""
+    # 📊 Dashboard Narrator - Powered by OpenRouter.ai
+    Unlock the hidden stories in your dashboards!<br>
+    Dashboard Narrator leverages advanced AI models through OpenRouter.ai to dissect your PDF reports,<br> 
+    analyze each segment with expert precision, and craft comprehensive insights in your preferred language.<br><br>
+    Turn complex data visualizations into clear, strategic recommendations and uncover trends you might have missed.<br>
+    From executive summaries to detailed breakdowns, get the full narrative behind your numbers in just a few clicks.<br><br>
+    """)
+    with gr.Row():
+        with gr.Column(scale=1):
+            api_key = gr.Textbox(label="OpenRouter API Key", placeholder="Enter your OpenRouter API key...", type="password")
+            refresh_btn = gr.Button("🔄 Refresh Available Models", size="sm")
+            
+            model_choice = gr.Dropdown(
+                choices=["meta-llama/llama-4-scout:free", "custom"] + OPENROUTER_MODELS,
+                value="meta-llama/llama-4-scout:free",
+                label="Select Model"
+            )
+            
+            custom_model = gr.Textbox(
+                label="Custom Model ID", 
+                placeholder="Enter custom model ID (e.g., anthropic/claude-3-opus:latest)...",
+                visible=False
+            )
+            
+            language = gr.Dropdown(
+                choices=["Italiano", "English", "Français", "Español", "Deutsch"], 
+                value="English", 
+                label="Report Language"
+            )
+            
+            num_sections = gr.Slider(
+                minimum=2, 
+                maximum=10, 
+                value=4, 
+                step=1, 
+                label="Number of Vertical Sections per Dashboard"
+            )
+            
+            goal = gr.Textbox(
+                label="Analysis Goal (optional)", 
+                placeholder="E.g., Analyze Q1 2024 sales KPIs..."
+            )
+            
+            pdf_files = gr.File(
+                label="Upload Dashboards (PDF)", 
+                file_types=[".pdf"], 
+                file_count="multiple"
+            )
+            
+            analyze_btn = gr.Button("🔍 Analyze Dashboards", variant="primary")
+            
+        with gr.Column(scale=2):
+            with gr.Tab("Report"):
+                output_md = gr.Markdown(label="Analysis Report", value="")
+            with gr.Tab("Output Files"):
+                output_files = gr.File(label="Download Files")
+            output_status = gr.Textbox(label="Progress", placeholder="Upload dashboards and press Analyze...", interactive=False)
+            # Progress component doesn't accept label in Gradio 5.21.0
+            progress_bar = gr.Progress()
+    
+    # Handle model dropdown change
+    model_choice.change(
+        fn=toggle_custom_model,
+        inputs=model_choice,
+        outputs=custom_model,
+    )
+    
+    # Handle refresh models button
+    refresh_btn.click(
+        fn=refresh_models,
+        inputs=api_key,
+        outputs=model_choice,
+    )
+    
+    # Handle analyze button
+    analyze_btn.click(
+        fn=process_dashboard,
+        inputs=[api_key, pdf_files, language, goal, num_sections, model_choice, custom_model],
+        outputs=[output_md, output_files, output_status]
+    )
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()