Update app.py

app.py

@@ -157,15 +157,13 @@ def interact_with_agent(file_input, additional_notes):
     os.makedirs(FIGURES_DIR, exist_ok=True)
 
     if file_input is None:
-        yield [{"role": "assistant", "content": "❌ No file uploaded. Please upload a CSV file to proceed."}]
-        return
+        return [{"role": "assistant", "content": "❌ No file uploaded. Please upload a CSV file to proceed."}]
 
     # Analyze the data
     data_description, visualization_paths, target = analyze_data(file_input.name)
 
     if data_description is None:
-        yield [{"role": "assistant", "content": data_description}]  # data_description contains the error message
-        return
+        return [{"role": "assistant", "content": data_description}]  # data_description contains the error message
 
     # Construct the prompt for the model
     prompt = base_prompt.format(
@@ -197,7 +195,7 @@ def interact_with_agent(file_input, additional_notes):
         else:
             messages.append({"role": "assistant", "content": f"⚠️ Unable to find image: {image_path}"})
 
-    yield messages
+    return messages
 
 # Define the Gradio interface
 with gr.Blocks(
@@ -248,3 +246,6 @@ if __name__ == "__main__":
 
 
 
+
+
+