Update app.py

app.py

@@ -8,51 +8,59 @@ def load_data(file):
     global df_global
     if file is None:
         return pd.DataFrame(), "Please upload a CSV file."
-    df_global = pd.read_csv(file.name)
-    preview = df_global.head(10)
-    return preview, f"Loaded {len(df_global)} rows and {len(df_global.columns)} columns."
+    try:
+        df_global = pd.read_csv(file.name)
+        preview = df_global.head(10)
+        return preview, f"✅ Loaded {len(df_global)} rows and {len(df_global.columns)} columns."
+    except Exception as e:
+        return pd.DataFrame(), f"❌ Error loading file: {e}"
 
 def filter_data(column, value):
     global df_global
     if df_global is None:
-        return pd.DataFrame(), "No data loaded."
-    if column == "" or value == "":
+        return pd.DataFrame(), "⚠️ No data loaded yet."
+
+    if not column or column not in df_global.columns:
+        return pd.DataFrame(), "⚠️ Please select a valid column."
+
+    if value.strip() == "":
         filtered = df_global
     else:
-        # Filter rows where the column contains the value (case-insensitive)
         filtered = df_global[df_global[column].astype(str).str.contains(value, case=False, na=False)]
+
     preview = filtered.head(10)
     summary = filtered.describe(include='all').to_string()
     return preview, summary
 
+def update_columns(file):
+    if file is None:
+        return []
+    try:
+        df = pd.read_csv(file.name)
+        return list(df.columns)
+    except:
+        return []
+
 with gr.Blocks() as demo:
-    gr.Markdown("# CSV Data Explorer\nUpload a CSV, filter data, and see summary stats.")
+    gr.Markdown("# 📊 CSV Data Explorer\nUpload a CSV file, filter it, and view summary statistics.")
 
     with gr.Row():
-        csv_file = gr.File(label="Upload CSV", file_types=['.csv'])
-        load_btn = gr.Button("Load Data")
+        csv_file = gr.File(label="Upload CSV", file_types=[".csv"])
+        load_btn = gr.Button("📥 Load Data")
 
-    table = gr.DataFrame(headers=None, interactive=True, label="Data Preview")
+    table = gr.DataFrame(headers=None, interactive=True, label="🔍 Data Preview")
     status = gr.Textbox(label="Status", interactive=False)
 
     with gr.Row():
-        filter_column = gr.Dropdown(choices=[], label="Filter Column", interactive=True)
-        filter_value = gr.Textbox(placeholder="Filter value", label="Filter Value")
-        filter_btn = gr.Button("Filter Data")
+        filter_column = gr.Dropdown(choices=[], label="Filter Column", interactive=True, allow_custom_value=False)
+        filter_value = gr.Textbox(placeholder="Enter value to search...", label="Filter Value")
+        filter_btn = gr.Button("🔎 Filter Data")
 
-    summary_stats = gr.Textbox(label="Summary Statistics", interactive=False, lines=10)
+    summary_stats = gr.Textbox(label="📈 Summary Statistics", interactive=False, lines=10)
 
+    # Link buttons to functions
     load_btn.click(load_data, inputs=csv_file, outputs=[table, status])
-
-    # Update filter_column choices dynamically when data loads
-    def update_columns(file):
-        if file is None:
-            return []
-        df = pd.read_csv(file.name)
-        return list(df.columns)
-
     csv_file.change(update_columns, inputs=csv_file, outputs=filter_column)
-
     filter_btn.click(filter_data, inputs=[filter_column, filter_value], outputs=[table, summary_stats])
 
 demo.launch()