stay hard

controller.py

@@ -27,6 +27,7 @@ import matplotlib
 import seaborn as sns
 from intitial_q_handler import if_initial_chart_question, if_initial_chat_question
 from rethink_gemini_agents.gemini_langchain_service import langchain_gemini_csv_chat
+from rethink_gemini_agents.rethink_chart import gemini_llm_chart
 from rethink_gemini_agents.rethink_chat import gemini_llm_chat
 from util_service import _prompt_generator, process_answer
 from fastapi.middleware.cors import CORSMiddleware
@@ -797,6 +798,18 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
         
         loop = asyncio.get_running_loop()
         # First, try the langchain-based method if the question qualifies
+        
+        if if_initial_chat_question(query):
+            answer = await asyncio.to_thread(
+                langchain_gemini_csv_chat, csv_url, query, False
+            )
+            logger.info("gemini langchain_answer --> ", answer)
+            return {"answer": jsonable_encoder(answer)}
+        
+        gemini_answer = await asyncio.to_thread(gemini_llm_chart, csv_url, query)
+        logger.info("gemini_answer --> ", gemini_answer)
+        return {"answer": gemini_answer}
+    
         if if_initial_chart_question(query):
             langchain_result = await loop.run_in_executor(
                 process_executor, langchain_csv_chart, csv_url, query, True