Update app.py

app.py

@@ -44,12 +44,14 @@ def clean_data(df):
 def upload_file(file):
     global df_global
     if file is None:
-        return pd.DataFrame({"Error": ["No file uploaded."]})
+        return pd.DataFrame({"Error": ["No file uploaded."]}), gr.update(choices=[])
     ext = os.path.splitext(file.name)[-1]
     df = pd.read_csv(file.name) if ext == ".csv" else pd.read_excel(file.name)
     df = clean_data(df)
     df_global = df
-    return df.head()
+    return df.head(), gr.update(choices=df.columns.tolist())
+
+
 
 def set_target_column(col_name):
     global target_column_global
@@ -387,7 +389,7 @@ with gr.Blocks() as demo:
             target_dropdown = gr.Dropdown(label="Select Target Column", choices=[], interactive=True)
             target_status = gr.Textbox(label="Target Column Status", interactive=False)
 
-            file_input.change(fn=upload_file, inputs=file_input, outputs=df_output)
+            file_input.change(fn=upload_file, inputs=file_input, outputs=[df_output, target_dropdown])
             file_input.change(fn=update_target_choices, inputs=[], outputs=target_dropdown)
             target_dropdown.change(fn=set_target_column, inputs=target_dropdown, outputs=target_status)