Fixing - TypeError: stream_to_gradio() got an unexpected keyword argument 'data_file'

app.py

@@ -16,7 +16,7 @@ model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct")
 agent = CodeAgent(
     tools=[],
     model=model,
-    additional_authorized_imports=["numpy", "pandas", "matplotlib.pyplot", "seaborn", "scipy.stats"],
+    additional_authorized_imports=["numpy", "pandas", "matplotlib.pyplot", "seaborn", "scipy.stats", "os", "shutil",],
     max_steps=10,
 )
 
@@ -77,14 +77,15 @@ def interact_with_agent(file_input, additional_notes):
     if additional_notes and len(additional_notes) > 0:
         prompt += "\nAdditional notes on the data:\n" + additional_notes
 
-    messages = [gr.ChatMessage(role="user", content=prompt)]
-    yield messages + [
-        gr.ChatMessage(role="assistant", content="⏳ _Starting task..._")
-    ]
+    prompt += "\n\nNOTE: The data is already loaded as 'data_file' in the Python interpreter. Use this directly without reloading."
 
+    messages = [gr.ChatMessage(role="user", content=prompt)]
+    yield messages + [gr.ChatMessage(role="assistant", content="⏳ _Starting task..._")]
     plot_image_paths = {}
-    for msg in stream_to_gradio(agent, prompt, data_file=data_file):
+    additional_args = {"data_file": data_file}
+    for msg in stream_to_gradio(agent, prompt, additional_args=additional_args):
         messages.append(msg)
+
         for image_path in get_images_in_directory("./figures"):
             if image_path not in plot_image_paths:
                 image_message = gr.ChatMessage(
@@ -93,9 +94,11 @@ def interact_with_agent(file_input, additional_notes):
                 )
                 plot_image_paths[image_path] = True
                 messages.append(image_message)
+
         yield messages + [
             gr.ChatMessage(role="assistant", content="⏳ _Still processing..._")
         ]
+
     yield messages