Update app.py

app.py

@@ -8,6 +8,46 @@ import sweetviz as sv
 from autoviz.AutoViz_Class import AutoViz_Class
 from traceml.summary.df import DataFrameSummary
 
+def variable_table(df):
+  """
+  Analyzes a DataFrame and categorizes variables with colorful HTML formatting.
+
+  Args:
+      df (pandas.DataFrame): The DataFrame to analyze.
+
+  Returns:
+      str: HTML code representing the analysis results with colorful highlights.
+  """
+  # Analyze variable types
+  categorical_vars = df.select_dtypes(include=['category', 'object']).columns.tolist()
+  numerical_vars = df.select_dtypes(include=['int64', 'float64']).columns.tolist()
+  text_vars = df.select_dtypes(include=['object']).difference(categorical_vars).tolist()
+
+  # Build HTML table with styles
+  table_style = 'border: 1px solid #ddd; border-collapse: collapse; text-align: left; font-size: 14px;'
+  header_style = 'background-color: #f2f2f2; padding: 5px 10px;'
+  data_style = 'padding: 5px 10px; border-bottom: 1px solid #ddd;'
+  category_color = '#90ee90'  # Light green for categorical
+  numerical_color = '#add8e6'  # Light blue for numerical
+  text_color = '#ffd9b3'  # Light yellow for text
+
+  html = f"<table style='{table_style}'>"
+  html += f"<tr><th style='{header_style}'>Variable Type</th><th style='{header_style}'>Columns</th></tr>"
+
+  # Add rows for each variable type with coloring
+  if categorical_vars:
+    html += f"<tr style='background-color: {category_color};'><td>Categorical</td><td style='{data_style}'>{', '.join(categorical_vars)}</td></tr>"
+  if numerical_vars:
+    html += f"<tr style='background-color: {numerical_color};'><td>Numerical</td><td style='{data_style}'>{', '.join(numerical_vars)}</td></tr>"
+  if text_vars:
+    html += f"<tr style='background-color: {text_color};'><td>Text</td><td style='{data_style}'>{', '.join(text_vars)}</td></tr>"
+
+  # Handle cases where no variables are found
+  if not (categorical_vars or numerical_vars or text_vars):
+    html += "<tr><td>No variables found!</td></tr>"
+
+  html += "</table>"
+  return html
 
 def generate_report(file, type):
     df = pd.read_csv(file) if file.name.endswith(".csv") else pd.read_excel(file)
@@ -87,7 +127,7 @@ def generate_report(file, type):
 
 
     
-    return temp_file1.name  ,temp_file3.name ,dfviz,sd1
+    return temp_file1.name  ,temp_file3.name ,variable_table(df),dfviz,sd1
     
      
 with gr.Blocks() as cluster:
@@ -107,11 +147,14 @@ with gr.Blocks() as cluster:
             gr.HTML(value="""<h1 style="color: #3399FF; text-shadow: 1px 1px 2px #ddd;">SWEETVIZ REPORT</h1>""")
             out3=gr.File(label="Download CSV")
         with gr.Column():
+            gr.Markdown("***Uploaded File***")
+            var=gr.HTML()
+        with gr.Column():
             
-            gr.Markdown("Uploaded File")
+            gr.Markdown("#Uploaded File")
             dataframe1=gr.Dataframe()
         with gr.Column():
-            gr.Markdown("Columns Analysis")
+            gr.Markdown("#Columns Analysis")
             dataframe2=gr.Dataframe()
-    btn.click(generate_report,inputs=[file],outputs=[out1,out3,dataframe1,dataframe2])
+    btn.click(generate_report,inputs=[file],outputs=[out1,out3,var,dataframe1,dataframe2])
 cluster.launch()