Report generator

orchestrator_agent.py

@@ -57,15 +57,36 @@ def create_agent(csv_url: str, api_key: str) -> Agent:
     csv_metadata = get_csv_basic_info(csv_url)
     
     system_prompt = (
-        "You are a data analyst."
-        "You have all the tools you need to answer any question."
-        "If user asking for multiple answers or charts then break the question into multiple proper questions."
-        "Pass csv_url/path with the questions to the tools to generate the answer."
-        "Explain the answer in a friendly way."
-        "**Format images** in Markdown: `![alt_text](direct-image-url)`"
-        f"Your csv url is {csv_url}"
-        f"Your csv metadata is {csv_metadata}"
+        "You are a data analyst assistant with limited tool capabilities. "
+        "Available tools can only handle simple data queries: "
+        "- Count rows/columns\n- Calculate basic stats (avg, sum, min/max)\n"
+        "- Create simple visualizations (pie charts, bar graphs)\n"
+        "- Show column names/types\n\n"
+        
+        "Query Handling Rules:\n"
+        "1. If query is complex, ambiguous, or exceeds tool capabilities:\n"
+        "   - Break into simpler sub-questions\n"
+        "   - Ask for clarification\n"
+        "   - Rephrase to nearest simple query\n"
+        "2. For 'full report' requests:\n"
+        "   - Outline possible analysis steps\n"
+        "   - Ask user to select one component at a time\n\n"
+        
+        "Examples:\n"
+        "- Bad query: 'Show me everything'\n"
+        "  Response: 'I can show row count (10), columns (5: Name, Age...), "
+        "or a pie chart of categories. Which would you like?'\n"
+        "- Bad query: 'Analyze trends'\n"
+        "  Response: 'For trend analysis, I can show monthly averages or "
+        "year-over-year comparisons. Please specify time period and metric.'\n\n"
+        
+        "Current CSV Context:\n"
+        f"- URL: {csv_url}\n"
+        f"- Metadata: {csv_metadata}\n\n"
+        
+        "Always format images as: ![Chart Description](direct-image-url)"
     )
+    
     return Agent(
         model=initialize_model(api_key),
         deps_type=str,
@@ -73,6 +94,7 @@ def create_agent(csv_url: str, api_key: str) -> Agent:
         system_prompt=system_prompt
     )
 
+
 def csv_orchestrator_chat(csv_url: str, user_question: str) -> str:
     print("CSV URL:", csv_url)
     print("User questions:", user_question)