Update app.py

app.py

@@ -3,7 +3,6 @@ import pandas as pd
 import sweetviz as sv
 import tempfile
 import os
-import re
 
 class DataAnalyzer:
     def __init__(self):
@@ -17,42 +16,21 @@ class DataAnalyzer:
         with open(report_path, 'r', encoding='utf-8') as f:
             html_content = f.read()
         
-        # Remove Sweetviz logo and header
-        html_content = re.sub(r'<div class="header">.*?</div>', '', html_content, flags=re.DOTALL)
-        html_content = re.sub(r'<div class="logo">.*?</div>', '', html_content, flags=re.DOTALL)
-        html_content = re.sub(r'<div class="nav-logo">.*?</div>', '', html_content, flags=re.DOTALL)
-        
-        # Additional CSS to hide unwanted elements and improve layout
-        custom_css = """
-        <style>
-        .header, .logo, .nav-logo {
-            display: none !important;
-        }
-        .container {
-            width: 100% !important;
-            max-width: none !important;
-            padding: 0 !important;
-            margin-top: 0 !important;
-        }
-        .nav {
-            padding-top: 0 !important;
-        }
-        .content {
-            padding-top: 0 !important;
-        }
-        .tab-content {
-            margin-top: 0 !important;
-        }
-        </style>
+        # Wrap the report in a table cell with styling
+        html_with_table = f"""
+        <table width="100%" style="border-collapse: collapse;">
+            <tr>
+                <td style="padding: 20px; border: 1px solid #ddd;">
+                    <div style="height: 800px; overflow: auto;">
+                        {html_content}
+                    </div>
+                </td>
+            </tr>
+        </table>
         """
         
-        # Insert custom CSS before closing head tag
-        html_content = html_content.replace('</head>', f'{custom_css}</head>')
-        
-        # Clean up
         os.remove(report_path)
-        
-        return html_content
+        return html_with_table
 
 def create_interface():
     analyzer = DataAnalyzer()
@@ -60,8 +38,13 @@ def create_interface():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("# Data Analysis Dashboard")
         
-        file_input = gr.File(label="Upload CSV")
-        report_html = gr.HTML()
+        with gr.Tabs():
+            with gr.TabItem("Sweetviz Analysis"):
+                file_input = gr.File(label="Upload CSV")
+                report_html = gr.HTML()
+            
+            with gr.TabItem("Custom Analysis"):
+                gr.Markdown("Custom analysis will be added here")
         
         def process_file(file):
             if file is None:
@@ -69,17 +52,7 @@ def create_interface():
             
             try:
                 df = pd.read_csv(file.name)
-                report = analyzer.generate_sweetviz_report(df)
-                
-                # Wrap report in container with fixed height
-                report_with_style = f"""
-                <div style="height: 800px; overflow: auto; padding: 0;">
-                    {report}
-                </div>
-                """
-                
-                return report_with_style
-                
+                return analyzer.generate_sweetviz_report(df)
             except Exception as e:
                 return f"Error generating report: {str(e)}"