Update app.py

app.py

@@ -8,6 +8,7 @@ from autogen.coding import LocalCommandLineCodeExecutor
 WORK_DIR = "coding"
 CSV_PATH = os.path.join(WORK_DIR, "input.csv")
 PLOT_PATH = os.path.join(WORK_DIR, "plot.png")
+CODE_PATH = os.path.join(WORK_DIR, "generated_code.py")
 os.makedirs(WORK_DIR, exist_ok=True)
 
 # Agents
@@ -20,6 +21,7 @@ code_writer_agent = AssistantAgent(
     system_message=(
         "You are a coding assistant. Write Python code that reads 'input.csv' in the current directory, "
         "performs analysis based on the user prompt, and saves the resulting plot to 'plot.png'. "
+        "If plot is not possible, output text or tabular results to 'results.txt'. "
         "Only generate the code, no explanations or comments."
     ),
 )
@@ -40,20 +42,37 @@ def process_csv_and_prompt(prompt, file):
     if file is not None:
         shutil.copyfile(file.name, CSV_PATH)
 
-    # Clean up old plot
+    # Clean up old outputs
     if os.path.exists(PLOT_PATH):
         os.remove(PLOT_PATH)
+    if os.path.exists(CODE_PATH):
+        os.remove(CODE_PATH)
+    results_path = os.path.join(WORK_DIR, "results.txt")
+    if os.path.exists(results_path):
+        os.remove(results_path)
 
     # Run agents
     code_executor_agent.reset()
     code_writer_agent.reset()
     full_prompt = f"{prompt}\n(Remember: the file is 'input.csv' in current directory)"
-    code_executor_agent.initiate_chat(code_writer_agent, message=full_prompt)
+    chat = code_executor_agent.initiate_chat(code_writer_agent, message=full_prompt)
 
-    if os.path.exists(PLOT_PATH):
-        return PLOT_PATH
+    # Save generated code
+    if chat and "content" in chat.messages[-1]:
+        generated_code = chat.messages[-1]["content"]
+        with open(CODE_PATH, "w") as f:
+            f.write(generated_code)
     else:
-        return None
+        generated_code = "No code captured."
+
+    # Decide outputs
+    plot = PLOT_PATH if os.path.exists(PLOT_PATH) else None
+    table_text = None
+    if os.path.exists(results_path):
+        with open(results_path, "r") as f:
+            table_text = f.read()
+
+    return plot, generated_code, table_text or "No tabular/text results produced."
 
 # Gradio Interface
 iface = gr.Interface(
@@ -62,9 +81,13 @@ iface = gr.Interface(
         gr.Textbox(label="Enter Analysis Prompt", placeholder="e.g. Generate 1000 Random numbers and plot histogram"),
         gr.File(label="Upload CSV", file_types=[".csv"]),
     ],
-    outputs=gr.Image(label="Generated Plot"),
+    outputs=[
+        gr.Image(label="Generated Plot"),
+        gr.Code(label="Generated Code"),
+        gr.Textbox(label="Tabular/Text Results")
+    ],
     title="Multi agent coding example",
-    description="I am a multi agent system, I write code and run it for you, optionally upload CSV file data for analysis!",
+    description="Upload CSV + analysis prompt. System generates Python code, executes it, and shows results as plot, code, and/or text.",
     allow_flagging="never",
 )