ok

controller.py

@@ -740,20 +740,14 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
         #     return {"image_url": image_public_url}
             # return FileResponse(groq_result, media_type="image/png")
         
-        # logger.info("Trying cerebras ai...")
-        # result = await query_csv_agent_cerebras(csv_url, query, chat_id)
-        # logger.info("cerebras ai result ==>", result)
-        # if result is not None and result != "":
-        #  return {"orchestrator_response": jsonable_encoder(result)}
-        
-        logger.info("Trying groq ai...")
-        result = await query_csv_agent_groq(csv_url, query, chat_id)
-        logger.info("groq ai result ==>", result)
+        logger.info("Trying cerebras ai...")
+        result = await query_csv_agent_cerebras(csv_url, query, chat_id)
+        logger.info("cerebras ai result ==>", result)
         if result is not None and result != "":
          return {"orchestrator_response": jsonable_encoder(result)}
           
         # Fallback: try langchain-based again
-        logger.error("Groq ai response failed, trying langchain groq....")
+        logger.error("Cerebras ai llama response failed, trying langchain groq....")
         langchain_paths = await loop.run_in_executor(
             process_executor, langchain_csv_chart, csv_url, query, True
         )
@@ -765,6 +759,7 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
            logger.info("Image uploaded to Supabase and Image URL is... ", {image_public_url})
            os.remove(langchain_paths[0])
            return {"image_url": image_public_url}
+           return FileResponse(langchain_paths[0], media_type="image/png")
         else:
            logger.error("All chart generation methods failed")
            return {"answer": "error"}