Update app.py

app.py

@@ -1,37 +1,15 @@
 import gradio as gr
 import pandas as pd
-from ydata_profiling import ProfileReport
 import sweetviz as sv
-from dataprep.eda import create_report
 import io
 
 class DataAnalyzer:
-    def __init__(self):
-        self.current_df = None
-    
-    def generate_profile_report(self, df, minimal=False):
-        profile = ProfileReport(
-            df,
-            minimal=minimal,
-            title="Pandas Profiling Report",
-            explorative=True,
-            dark_mode=True
-        )
-        # Get HTML directly as string
-        return profile.to_html()
-    
     def generate_sweetviz_report(self, df):
         report = sv.analyze(df)
-        # Use StringIO to capture the HTML output
         html_io = io.StringIO()
         report.show_html(filepath=html_io, open_browser=False)
         return html_io.getvalue()
     
-    def generate_dataprep_report(self, df):
-        report = create_report(df)
-        # Get HTML directly as string
-        return report.html()
-    
     def get_dataset_info(self, df):
         return {
             "Rows": len(df),
@@ -45,102 +23,39 @@ def create_interface():
     analyzer = DataAnalyzer()
     
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown("""
-        # Data Analysis Dashboard
-        Upload your CSV file to generate interactive analysis reports
-        """)
+        gr.Markdown("# Data Analysis Dashboard")
         
         with gr.Row():
             file_input = gr.File(label="Upload CSV")
             dataset_info = gr.JSON(label="Dataset Information")
         
-        with gr.Row():
-            report_type = gr.Radio(
-                choices=["Full", "Minimal"],
-                value="Full",
-                label="Report Type"
-            )
-        
         with gr.Tabs():
-            with gr.TabItem("Pandas Profiling"):
-                profile_html = gr.HTML()
-            with gr.TabItem("Sweetviz"):
+            with gr.TabItem("Sweetviz Report"):
                 sweet_html = gr.HTML()
-            with gr.TabItem("DataPrep"):
-                prep_html = gr.HTML()
+                download_btn = gr.Button("Download Report")
         
-        def process_file(file, report_type):
+        def process_file(file):
             if file is None:
-                return None, None, None, None
+                return None, None
             
             try:
                 df = pd.read_csv(file.name)
-                analyzer.current_df = df
-                
-                # Get dataset info
                 info = analyzer.get_dataset_info(df)
+                report_html = analyzer.generate_sweetviz_report(df)
                 
-                # Generate reports
-                minimal = report_type == "Minimal"
-                
-                with gr.Progress() as progress:
-                    progress(0, desc="Generating Pandas Profiling report...")
-                    profile_html = analyzer.generate_profile_report(df, minimal)
-                    
-                    progress(0.33, desc="Generating Sweetviz report...")
-                    sweet_html = analyzer.generate_sweetviz_report(df)
-                    
-                    progress(0.66, desc="Generating DataPrep report...")
-                    prep_html = analyzer.generate_dataprep_report(df)
-                    
-                    progress(1.0, desc="Done!")
-                
-                return (
-                    info,
-                    profile_html,
-                    sweet_html,
-                    prep_html
-                )
+                return info, report_html
                 
             except Exception as e:
-                return str(e), None, None, None
+                return str(e), None
         
         file_input.change(
             fn=process_file,
-            inputs=[file_input, report_type],
-            outputs=[dataset_info, profile_html, sweet_html, prep_html]
-        )
-        
-        report_type.change(
-            fn=process_file,
-            inputs=[file_input, report_type],
-            outputs=[dataset_info, profile_html, sweet_html, prep_html]
+            inputs=[file_input],
+            outputs=[dataset_info, sweet_html]
         )
     
     return demo
 
-# Add custom CSS for better HTML rendering
-custom_css = """
-<style>
-.report-container {
-    width: 100%;
-    height: 800px;
-    overflow: auto;
-}
-.report-container iframe {
-    width: 100%;
-    height: 100%;
-    border: none;
-}
-</style>
-"""
-
-# Launch the interface
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch(
-        share=True,  # Enable sharing
-        height=1000,  # Set interface height
-        show_error=True,  # Show detailed error messages
-        custom_css=custom_css  # Apply custom styling
-    )
+    demo.launch()