Create app.py

app.py

@@ -0,0 +1,72 @@
+import os
+import shutil
+import gradio as gr
+from autogen import ConversableAgent, AssistantAgent
+from autogen.coding import LocalCommandLineCodeExecutor
+
+# Setup
+WORK_DIR = "coding"
+CSV_PATH = os.path.join(WORK_DIR, "input.csv")
+PLOT_PATH = os.path.join(WORK_DIR, "plot.png")
+os.makedirs(WORK_DIR, exist_ok=True)
+
+# Agents
+llm_config = {"model": "gpt-4-turbo"}
+
+code_writer_agent = AssistantAgent(
+    name="code_writer_agent",
+    llm_config=llm_config,
+    human_input_mode="NEVER",
+    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'. "
+        "Only generate the code, no explanations or comments."
+    ),
+)
+
+executor = LocalCommandLineCodeExecutor(timeout=60, work_dir=WORK_DIR)
+
+code_executor_agent = ConversableAgent(
+    name="code_executor_agent",
+    llm_config=False,
+    code_execution_config={"executor": executor},
+    human_input_mode="NEVER",
+    default_auto_reply="TERMINATE",
+)
+
+# Main function
+def process_csv_and_prompt(prompt, file):
+    # Save uploaded file
+    if file is not None:
+        shutil.copyfile(file.name, CSV_PATH)
+
+    # Clean up old plot
+    if os.path.exists(PLOT_PATH):
+        os.remove(PLOT_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)
+
+    if os.path.exists(PLOT_PATH):
+        return PLOT_PATH
+    else:
+        return None
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=process_csv_and_prompt,
+    inputs=[
+        gr.Textbox(label="Enter Analysis Prompt", placeholder="e.g. Plot average of column B by category"),
+        gr.File(label="Upload CSV", file_types=[".csv"]),
+    ],
+    outputs=gr.Image(label="Generated Plot"),
+    title="CSV Plotter & Code Runner",
+    description="I am a multi agent system, I write code and run it for you, optionally upload CSV file data for analysis!",
+    allow_flagging="never",
+)
+
+if __name__ == "__main__":
+    iface.launch()