Update app.py

app.py

@@ -2,8 +2,12 @@ import gradio as gr
 import pandas as pd
 import sweetviz as sv
 import io
+import base64
 
 class DataAnalyzer:
+    def __init__(self):
+        self.current_df = None
+    
     def generate_sweetviz_report(self, df):
         report = sv.analyze(df)
         html_io = io.StringIO()
@@ -11,47 +15,49 @@ class DataAnalyzer:
         return html_io.getvalue()
     
     def get_dataset_info(self, df):
-        return {
+        info_dict = {
             "Rows": len(df),
             "Columns": len(df.columns),
-            "Memory Usage (MB)": df.memory_usage(deep=True).sum() / 1024**2,
-            "Missing Values": df.isnull().sum().sum(),
-            "Data Types": df.dtypes.value_counts().to_dict()
+            "Memory Usage (MB)": round(df.memory_usage(deep=True).sum() / 1024**2, 2),
+            "Missing Values": int(df.isnull().sum().sum()),
+            "Data Types": df.dtypes.astype(str).to_dict()
         }
+        return str(info_dict)  # Convert to string to avoid JSON serialization issues
 
 def create_interface():
     analyzer = DataAnalyzer()
     
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown("# Data Analysis Dashboard")
+        gr.Markdown("# CSV Data Analysis Dashboard")
         
         with gr.Row():
             file_input = gr.File(label="Upload CSV")
-            dataset_info = gr.JSON(label="Dataset Information")
+            info_output = gr.Textbox(label="Dataset Information", lines=10)
         
-        with gr.Tabs():
-            with gr.TabItem("Sweetviz Report"):
-                sweet_html = gr.HTML()
-                download_btn = gr.Button("Download Report")
+        report_html = gr.HTML()
         
         def process_file(file):
             if file is None:
-                return None, None
+                return "No file uploaded.", None
             
             try:
                 df = pd.read_csv(file.name)
                 info = analyzer.get_dataset_info(df)
-                report_html = analyzer.generate_sweetviz_report(df)
+                report = analyzer.generate_sweetviz_report(df)
+                
+                # Create download link
+                b64 = base64.b64encode(report.encode()).decode()
+                download_link = f'<a href="data:text/html;base64,{b64}" download="analysis_report.html">Download Report</a>'
                 
-                return info, report_html
+                return info, report + download_link
                 
             except Exception as e:
-                return str(e), None
+                return f"Error: {str(e)}", None
         
         file_input.change(
             fn=process_file,
             inputs=[file_input],
-            outputs=[dataset_info, sweet_html]
+            outputs=[info_output, report_html]
         )
     
     return demo