stay hard

controller.py

@@ -26,9 +26,6 @@ import matplotlib.pyplot as plt
 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_chart, 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
 import matplotlib
@@ -297,55 +294,12 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
         decoded_url = unquote(csv_url)
 
         if if_initial_chat_question(query):
-            
-            
-             # --- Gemini-Langchain-based and Rethink-based chat generation starts---
-            
-            langchain_gemini_answer = await asyncio.to_thread(
-                langchain_gemini_csv_chat, decoded_url, query, True
-            )
-            logger.info("Gemini langchain_answer:", langchain_gemini_answer)
-            if langchain_gemini_answer != None and process_answer(langchain_gemini_answer) != True: 
-             return {"answer": jsonable_encoder(langchain_gemini_answer)}
-        
-            # --- Gemini-Langchain-based and Rethink-based chat generation ends---
-            
-            
-            
             answer = await asyncio.to_thread(
                 langchain_csv_chat, decoded_url, query, False
             )
-            logger.info("Groq langchain_answer:", answer)
-            if answer != None and process_answer(answer) != True:
-             return {"answer": jsonable_encoder(answer)} 
-        
-        
-          # --- Gemini-Langchain-based and Rethink-based chat generation starts---
-            
-            
-            answer = await asyncio.to_thread(
-                gemini_llm_chat, decoded_url, query
-            )
-            logger.info("Rethink gemini 1st:", answer)
+            logger.info("langchain_answer:", answer)
             return {"answer": jsonable_encoder(answer)}
-        
-            
-          # --- Gemini-Langchain-based and Rethink-based chat generation ends---
-            
-            
-        # --- Gemini-Langchain-based and Rethink-based chat generation starts---
-            
-            
-        answer = await asyncio.to_thread(
-            gemini_llm_chat, decoded_url, query
-        )
-        logger.info("Rethink gemini 1st:", answer)
-        if(answer != None and process_answer(answer) != True):
-           return {"answer": jsonable_encoder(answer)}
-        
-            
-          # --- Gemini-Langchain-based and Rethink-based chat generation ends---
-            
+
         # Process with groq_chat first
         groq_answer = await asyncio.to_thread(groq_chat, decoded_url, query)
         logger.info("groq_answer:", groq_answer)
@@ -358,20 +312,7 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
                 langchain_csv_chat, decoded_url, query, False
             )
             if process_answer(lang_answer):
-               logger.error("Error in Groq_Langchain, Trying Gemini...")
-               
-               # --- Gemini-Langchain-based and Rethink-based chat generation starts---
-            
-               langchain_gemini_answer = await asyncio.to_thread(
-                   langchain_gemini_csv_chat, decoded_url, query, True
-               )
-               logger.info("Gemini langchain_answer:", langchain_gemini_answer)
-               if langchain_gemini_answer != None and process_answer(langchain_gemini_answer) != True: 
-                return {"answer": jsonable_encoder(langchain_gemini_answer)}
-        
-               # --- Gemini-Langchain-based and Rethink-based chat generation ends---
-            
-            
+                return {"answer": "error"}
             return {"answer": jsonable_encoder(lang_answer)}
 
         return {"answer": jsonable_encoder(groq_answer)}
@@ -843,51 +784,12 @@ 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_chart_question(query):
-            
-            
-            # --- Gemini-Langchain-based and Rethink-based chart generation starts here ---
-            
-            
-            langchain_gemini_chart_answer = await asyncio.to_thread(
-                langchain_gemini_csv_chart, csv_url, query, True
-            )
-            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")
-    
-            logger.error("Gemini Langchain chart generation failed for initial question, trying groq langchain....")
-            
-            
-   
-            
-            # --- Gemini-Langchain-based and Rethink-based chart generation ends here ---
-            
-            
             langchain_result = await loop.run_in_executor(
                 process_executor, langchain_csv_chart, csv_url, query, True
             )
             logger.info("Langchain chart result:", langchain_result)
             if isinstance(langchain_result, list) and len(langchain_result) > 0:
                return FileResponse(langchain_result[0], media_type="image/png")
-           
-           
-           
-               
-        # --- Gemini-Langchain-based and Rethink-based chart generation starts here ---
-            
-            
-        logger.error("Groq Langchain chart generation failed, trying rethink gemini....")
-        gemini_answer = await asyncio.to_thread(gemini_llm_chart, csv_url, query)
-        logger.info("gemini_answer --> ", gemini_answer)
-        if(isinstance(gemini_answer, str) and gemini_answer != "Chart path not found" and ".png" in gemini_answer):
-          return FileResponse(gemini_answer, media_type="image/png")
-        
-        logger.error("Gemini chart generation failed, trying groq....")
-            
-            
-        # --- Gemini-Langchain-based and Rethink-based chart generation ends here ---
-        
-        
         
         # Next, try the groq-based method
         groq_result = await loop.run_in_executor(
@@ -906,21 +808,8 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
         if isinstance(langchain_paths, list) and len(langchain_paths) > 0:
            return FileResponse(langchain_paths[0], media_type="image/png")
         else:
-            
-            # --- Gemini-Langchain-based and Rethink-based chart generation starts here ---
-            
-            
-            langchain_gemini_chart_answer = await asyncio.to_thread(
-                langchain_gemini_csv_chart, csv_url, query, True
-            )
-            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-Langchain-based and Rethink-based chart generation ends here ---
-            
-            logger.error("All chart generation methods failed")
-            return {"error": "All chart generation methods failed"}
+           logger.error("All chart generation methods failed")
+           return {"error": "All chart generation methods failed"}
 
     except Exception as e:
         logger.error(f"Critical chart error: {str(e)}")