import gradio as gr
import pandas as pd
import sweetviz as sv
import tempfile
import os
import re

class DataAnalyzer:
    def __init__(self):
        self.temp_dir = tempfile.mkdtemp()
    
    def generate_sweetviz_report(self, df):
        report = sv.analyze(df)
        report_path = os.path.join(self.temp_dir, "report.html")
        report.show_html(report_path, open_browser=False)
        
        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>
        """
        
        # 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

def create_interface():
    analyzer = DataAnalyzer()
    
    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()
        
        def process_file(file):
            if file is None:
                return None
            
            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
                
            except Exception as e:
                return f"Error generating report: {str(e)}"
        
        file_input.change(
            fn=process_file,
            inputs=[file_input],
            outputs=[report_html]
        )
    
    return demo

if __name__ == "__main__":
    demo = create_interface()
    demo.launch(show_error=True)