Update app.py

app.py

@@ -1,18 +1,24 @@
 import gradio as gr
 import pandas as pd
 import sweetviz as sv
-import io
-import base64
+import tempfile
+import os
 
 class DataAnalyzer:
     def __init__(self):
+        self.temp_dir = tempfile.mkdtemp()
         self.current_df = None
     
     def generate_sweetviz_report(self, df):
         report = sv.analyze(df)
-        html_io = io.StringIO()
-        report.show_html(filepath=html_io, open_browser=False)
-        return html_io.getvalue()
+        # Save to temporary file
+        temp_path = os.path.join(self.temp_dir, "report.html")
+        report.show_html(temp_path, open_browser=False)
+        
+        # Read the content
+        with open(temp_path, 'r', encoding='utf-8') as f:
+            content = f.read()
+        return content
     
     def get_dataset_info(self, df):
         info_dict = {
@@ -20,9 +26,9 @@ class DataAnalyzer:
             "Columns": len(df.columns),
             "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()
+            "Column Types": df.dtypes.astype(str).to_dict()
         }
-        return str(info_dict)  # Convert to string to avoid JSON serialization issues
+        return info_dict
 
 def create_interface():
     analyzer = DataAnalyzer()
@@ -32,32 +38,28 @@ def create_interface():
         
         with gr.Row():
             file_input = gr.File(label="Upload CSV")
-            info_output = gr.Textbox(label="Dataset Information", lines=10)
+            dataset_info = gr.JSON(label="Dataset Information")
         
         report_html = gr.HTML()
         
         def process_file(file):
             if file is None:
-                return "No file uploaded.", None
+                return None, None
             
             try:
                 df = pd.read_csv(file.name)
                 info = analyzer.get_dataset_info(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 + download_link
+                return info, report
                 
             except Exception as e:
-                return f"Error: {str(e)}", None
+                return {"error": str(e)}, None
         
         file_input.change(
             fn=process_file,
             inputs=[file_input],
-            outputs=[info_output, report_html]
+            outputs=[dataset_info, report_html]
         )
     
     return demo