Update app.py

app.py

@@ -23,14 +23,27 @@ all_headers = config_dict["sklearn"]["columns"]
 headers = [col for col in all_headers if col in df.columns]
 
 # Define input and output interfaces
-inputs = [gr.Dataframe(headers=headers, row_count=(2, "dynamic"), col_count=(len(headers), "fixed"), label="Input Data", interactive=True)]
+#inputs = [gr.Dataframe(headers=headers, row_count=(2, "dynamic"), col_count=(len(headers), "fixed"), label="Input Data", interactive=True)]
+inputs = [gr.Dataframe(headers=all_headers, row_count=(2, "dynamic"), col_count=(len(all_headers), "fixed"), label="Input Data", interactive=True)]
 outputs = [gr.Dataframe(row_count=(2, "dynamic"), col_count=(1, "fixed"), label="Predictions", headers=["Depression"])]
 
+#def infer(inputs):
+    #data = pd.DataFrame(inputs, columns=headers)
+    #predictions = pipe.predict(data)
+    #return pd.DataFrame(predictions, columns=["Depression"])
+
 def infer(inputs):
     data = pd.DataFrame(inputs, columns=headers)
+    # Add missing columns with default values (e.g., 0)
+    for col in all_headers:
+        if col not in data.columns:
+            data[col] = 0
+    # Ensure the order of columns matches the training data
+    data = data[all_headers]
     predictions = pipe.predict(data)
     return pd.DataFrame(predictions, columns=["Depression"])
 
+
 gr.Interface(
     fn=infer,
     inputs=inputs,