stay hard

api_key_rotation.log

@@ -0,0 +1,7 @@
+2025-02-23 10:59:01,652 - INFO - Max CPUs: 12
+2025-02-23 10:59:16,924 - INFO - Successfully configured with key: AIzaSyC6...YN7s
+2025-02-23 10:59:19,466 - WARNING - Retry 1/5...
+2025-02-23 11:00:44,274 - INFO - Max CPUs: 12
+2025-02-23 11:02:10,466 - INFO - Successfully configured with key: AIzaSyC6...YN7s
+2025-02-23 11:13:51,950 - INFO - Successfully configured with key: AIzaSyC6...YN7s
+2025-02-23 11:16:15,194 - INFO - Successfully configured with key: AIzaSyC6...YN7s

controller.py

@@ -800,15 +800,16 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
         # First, try the langchain-based method if the question qualifies
         
         if if_initial_chat_question(query):
-            answer = await asyncio.to_thread(
+            langchain_gemini_chart_answer = await asyncio.to_thread(
                 langchain_gemini_csv_chat, csv_url, query, False
             )
-            logger.info("gemini langchain_answer --> ", answer)
-            return {"answer": jsonable_encoder(answer)}
+            logger.info("gemini langchain_answer --> ", langchain_gemini_chart_answer)
+            if isinstance(langchain_gemini_chart_answer, list) and len(langchain_gemini_chart_answer) > 0:
+               return FileResponse(langchain_gemini_chart_answer[0], media_type="image/png")
         
         gemini_answer = await asyncio.to_thread(gemini_llm_chart, csv_url, query)
         logger.info("gemini_answer --> ", gemini_answer)
-        return {"answer": gemini_answer}
+        return FileResponse(gemini_answer, media_type="image/png")
     
         if if_initial_chart_question(query):
             langchain_result = await loop.run_in_executor(

rethink_gemini_agents/gemini_langchain_service.py

@@ -197,16 +197,9 @@ def langchain_gemini_csv_chart(csv_url: str, question: str, chart_required: bool
 
 
 # Example usage:
-# if __name__ == "__main__":
-#     csv_url = "./documents/titanic.csv"
-#     question = "Create 2 beautiful visualizations of the data using different chart styles (line, bar etc..), return file names"
-#     output = langchain_gemini_csv_chat(csv_url, question, True)
-#     print("Agent output:", output)
-    
-#     # Define a regex pattern that matches 'temp' followed by one or more digits.
-#     pattern = r"temp\d+"
-
-#     # Use re.findall to extract all occurrences that match the pattern.
-#     names = re.findall(pattern, output)
-    
-#     print(names)
+if __name__ == "__main__":
+    csv_url = "./documents/titanic.csv"
+    question = "Create a pie chart of males vs females"
+    output = langchain_gemini_csv_chat(csv_url, question, True)
+    print("Agent output:", output)
+   

rethink_gemini_agents/rethink_chart.py

@@ -10,7 +10,6 @@ from typing import Optional, Dict, Any, List
 from pydantic import BaseModel, Field
 from google.generativeai import GenerativeModel, configure
 from dotenv import load_dotenv
-from csv_service import clean_data
 
 # Load environment variables from .env file
 load_dotenv()
@@ -211,7 +210,7 @@ class RethinkAgent(BaseModel):
 
 
 def gemini_llm_chart(csv_url: str, query: str) -> str:
-    df = clean_data(csv_url)
+    df = pd.read_csv(csv_url)
     
     agent = RethinkAgent(df=df)
     if not agent.initialize_model(API_KEYS):
@@ -221,6 +220,22 @@ def gemini_llm_chart(csv_url: str, query: str) -> str:
     result = agent.execute_query(query)
     print("\nAnalysis Result:")
     print(result)
+    
+    if isinstance(result, str):
+     result = result.strip()  # Remove any leading/trailing spaces or newlines
+
+     match = re.search(r'CHART_SAVED:\s*(\S+)', result)
+
+     if match:
+        chart_path = match.group(1)
+        print("Chart Path:", chart_path)
+        return chart_path
+     else:
+        print("Chart path not found")
+        return None
+    else:
+     print("Unexpected result format:", type(result))
+     return None
 
 
 
@@ -235,4 +250,17 @@ def gemini_llm_chart(csv_url: str, query: str) -> str:
         
 #     result = agent.execute_query("Create a scatter plot of total_bill vs tip with kernel density estimate")
 #     print("\nAnalysis Result:")
-#     print(result)
+#     print(result)
+    
+#     if isinstance(result, str):
+#      result = result.strip()  # Remove any leading/trailing spaces or newlines
+
+#      match = re.search(r'CHART_SAVED:\s*(\S+)', result)
+
+#      if match:
+#         chart_path = match.group(1)
+#         print("Chart Path:", chart_path)
+#      else:
+#         print("Chart path not found")
+#     else:
+#      print("Unexpected result format:", type(result))