Update app.py

app.py

@@ -51,6 +51,33 @@ def update_subdomains(domain, df):
     subdomains = df[df["domain"] == domain]["subdomain"].dropna().unique().tolist()
     return gr.Dropdown.update(choices=sorted(subdomains), value=None)
 
+# Load and handle file upload
+def load_and_update(file):
+    if file is not None:
+        try:
+            df = pd.read_csv(file.name)  # Read the CSV file
+            if df.empty:
+                return "⚠️ The uploaded dataset is empty."
+            
+            # Ensure expected columns exist
+            required_columns = ['domain', 'subdomain', 'keywords', 'question', 'option1', 'option2', 'option3', 'option4', 'correct_answer']
+            missing_columns = set(required_columns) - set(df.columns)
+            if missing_columns:
+                return f"⚠️ Missing required columns in the dataset: {', '.join(missing_columns)}"
+            
+            # Normalize text to avoid case/whitespace mismatches
+            df['domain'] = df['domain'].str.strip().str.lower()
+            df['subdomain'] = df['subdomain'].str.strip()
+
+            # Update domain dropdown choices
+            domains = sorted(df['domain'].unique().tolist())
+            domain_dropdown.update(choices=domains)
+
+            return df
+        except Exception as e:
+            return f"Error loading file: {str(e)}"
+    return None
+
 # Gradio UI
 def launch_interface():
     with gr.Blocks() as demo:
@@ -63,19 +90,6 @@ def launch_interface():
         file_upload = gr.File(label="Upload MCQ Dataset (CSV)", type="file")
 
         # Update domain list when file is uploaded
-        def load_and_update(file):
-            if file is not None:
-                df = pd.read_csv(file.name)
-                # Normalize text to avoid case/whitespace mismatches
-                df['domain'] = df['domain'].str.strip().str.lower()
-                df['subdomain'] = df['subdomain'].str.strip()
-                
-                domains = sorted(df['domain'].unique().tolist())
-                domain_dropdown.update(choices=domains)
-
-                return df
-            return None
-
         file_upload.change(fn=load_and_update, inputs=file_upload, outputs=None)
 
         domain_dropdown.change(fn=update_subdomains, inputs=domain_dropdown, outputs=subdomain_dropdown)