Update app.py

app.py

@@ -614,4 +614,174 @@ def process_multiple_dashboards(api_key, pdf_files, language_code="it", goal_des
     # Return the combined report content and all output files
     combined_content = "\n\n---\n\n".join(individual_reports)
     if len(individual_reports) > 1 and 'comparative_report' in locals():
-        combined_content += f"\n\n{'='*80}\n\n# COMPARATIVE ANALYSIS\n\n{comparative_report}"
+        combined_content += f"\n\n{'='*80}\n\n# COMPARATIVE ANALYSIS\n\n{comparative_report}"
+    
+    return combined_content, output_files, "✅ Analysis completed successfully!"
+
+# Gradio Interface Functions
+def process_dashboard(api_key, pdf_files, language_choice, analysis_goal, num_sections, model_choice, custom_model, progress=gr.Progress()):
+    """Process the dashboard via the Gradio interface."""
+    if not api_key or not pdf_files:
+        return None, None, "Please provide both the API key and at least one PDF file."
+    
+    # Map language choice to language code
+    language_codes = {
+        "Italiano": "it",
+        "English": "en",
+        "Français": "fr",
+        "Español": "es",
+        "Deutsch": "de"
+    }
+    language_code = language_codes.get(language_choice, "it")
+    
+    # Validate number of sections
+    try:
+        num_sections = int(num_sections)
+        if num_sections < 1:
+            num_sections = 4  # Default
+    except:
+        num_sections = 4  # Default
+    
+    # Convert the uploaded files to a list of bytes
+    progress(0, desc="Preparing files...")
+    pdf_bytes_list = []
+    for pdf_file in pdf_files:
+        with open(pdf_file.name, "rb") as f:
+            pdf_bytes_list.append(f.read())
+    
+    # Start progress thread for text updates
+    progress_thread = threading.Thread(target=update_progress)
+    progress_thread.daemon = True
+    progress_thread.start()
+    
+    # Process the dashboards
+    try:
+        # Override the progress_tracker update function to also update the Gradio progress bar
+        original_update = progress_tracker.update
+        
+        def combined_update(percent, message="Processing..."):
+            original_update(percent, message)
+            progress(percent/100, desc=message)
+            
+        progress_tracker.update = combined_update
+        
+        md_content, output_files, status_message = process_multiple_dashboards(
+            api_key=api_key,
+            pdf_files=pdf_bytes_list,
+            language_code=language_code,
+            goal_description=analysis_goal,
+            num_sections=num_sections,
+            model_name=model_choice,
+            custom_model=custom_model
+        )
+        
+        if md_content and output_files:
+            progress(1.0, desc="Complete!")
+            return md_content, output_files, progress_tracker.get_status()
+        else:
+            return None, None, "❌ Error during dashboard analysis."
+    except Exception as e:
+        progress_tracker.update(100, f"❌ Error: {str(e)}")
+        progress_tracker.end_processing()
+        progress(1.0, desc=f"Error: {str(e)}")
+        return None, None, f"❌ Error: {str(e)}"
+
+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()