Spaces:

shukdevdattaEX
/

Data-Summarizer-Excel-CSV

Paused

App Files Files Community

shukdevdattaEX commited on Sep 1, 2025

Commit

e556499

verified ·

1 Parent(s): 946d3e8

Update app.py

Browse files

Files changed (1) hide show

app.py +51 -12

app.py CHANGED Viewed

@@ -12,6 +12,8 @@ from typing import Optional, Tuple, Dict, Any
 import logging
 from datetime import datetime
 import re
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -325,6 +327,11 @@ def create_basic_charts(df: pd.DataFrame) -> str:
             fig.update_layout(height=400)
             charts_html.append(fig.to_html(include_plotlyjs='cdn'))
         return "\n".join(charts_html) if charts_html else "<p>No charts generated for this dataset.</p>"
     except Exception as e:
@@ -339,12 +346,12 @@ def clear_all():
     """Clear all inputs and outputs"""
     return None, "", "", "", "", "", None
-def download_summary(analysis_text, data_summary):
-    """Generate downloadable summary report"""
     if not analysis_text:
         return None
-    report = f"""# Data Analysis Report
 Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
 ## AI Analysis:
@@ -354,12 +361,34 @@ Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
 {data_summary}
 """
-    # Save to temporary file
-    filename = f"data_analysis_report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md"
-    with open(filename, 'w', encoding='utf-8') as f:
-        f.write(report)
-    return filename
 # Create enhanced Gradio interface
 with gr.Blocks(
@@ -467,7 +496,12 @@ with gr.Blocks(
         with gr.Tab("💾 Export"):
             gr.Markdown("### Download Your Analysis Report")
-            download_btn = gr.Button("📥 Download Report (.md)", variant="secondary")
             download_file = gr.File(label="Download Link", visible=False)
     # Event handlers
@@ -494,7 +528,7 @@ with gr.Blocks(
     ask_btn.click(
         fn=sync_analyze_data,
         inputs=[file_input, api_key_input, question_input],
-        outputs=[question_output, gr.Textbox(visible=False), gr.HTML(visible=False), gr.HTML(visible=False)],
         show_progress=True
     )
@@ -515,7 +549,7 @@ with gr.Blocks(
     # Download functionality
     download_btn.click(
         fn=download_summary,
-        inputs=[analysis_output, raw_summary],
         outputs=[download_file]
     )
@@ -541,5 +575,10 @@ with gr.Blocks(
 if __name__ == "__main__":
     app.queue(max_size=10)  # Handle multiple users
     app.launch(
-        share=True
     )

 import logging
 from datetime import datetime
 import re
+import markdown
+from weasyprint import HTML as WeasyHTML
 # Configure logging
 logging.basicConfig(level=logging.INFO)
             fig.update_layout(height=400)
             charts_html.append(fig.to_html(include_plotlyjs='cdn'))
+        # Additional charts from generate_chart_data
+        charts_data = analyzer.generate_chart_data(df)
+        for key, fig in charts_data.items():
+            charts_html.append(fig.to_html(include_plotlyjs='cdn'))
         return "\n".join(charts_html) if charts_html else "<p>No charts generated for this dataset.</p>"
     except Exception as e:
     """Clear all inputs and outputs"""
     return None, "", "", "", "", "", None
+def download_summary(analysis_text, data_summary, format_choice):
+    """Generate downloadable summary report in chosen format"""
     if not analysis_text:
         return None
+    report_md = f"""# Data Analysis Report
 Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
 ## AI Analysis:
 {data_summary}
 """
+    base_filename = f"data_analysis_report_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
+    filename = None
+    try:
+        if format_choice == "PDF":
+            # Convert MD to HTML first
+            report_html = markdown.markdown(report_md)
+            # Wrap in basic HTML structure for better PDF rendering
+            full_html = f"""
+            <html>
+            <head><style>body {{ font-family: Arial, sans-serif; }}</style></head>
+            <body>{report_html}</body>
+            </html>
+            """
+            filename = base_filename + ".pdf"
+            WeasyHTML(string=full_html).write_pdf(filename)
+        elif format_choice == "HTML":
+            report_html = markdown.markdown(report_md, output_format='html5')
+            filename = base_filename + ".html"
+            with open(filename, 'w', encoding='utf-8') as f:
+                f.write(report_html)
+        return filename
+    except Exception as e:
+        logger.error(f"Download generation error: {str(e)}")
+        return None
 # Create enhanced Gradio interface
 with gr.Blocks(
         with gr.Tab("💾 Export"):
             gr.Markdown("### Download Your Analysis Report")
+            format_choice = gr.Dropdown(
+                choices=["PDF", "HTML"],
+                label="Choose Format",
+                value="PDF"
+            )
+            download_btn = gr.Button("📥 Download Report", variant="secondary")
             download_file = gr.File(label="Download Link", visible=False)
     # Event handlers
     ask_btn.click(
         fn=sync_analyze_data,
         inputs=[file_input, api_key_input, question_input],
+        outputs=[question_output, gr.Textbox(visible=False), gr.HTML(visible=False), charts_output],  # Update charts on question too
         show_progress=True
     )
     # Download functionality
     download_btn.click(
         fn=download_summary,
+        inputs=[analysis_output, raw_summary, format_choice],
         outputs=[download_file]
     )
 if __name__ == "__main__":
     app.queue(max_size=10)  # Handle multiple users
     app.launch(
+        share=True,
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True,
+        favicon_path=None,
+        ssl_verify=False
     )