Successful Truncation

controller.py

@@ -399,33 +399,33 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
             logger.info("Starting report generation process...")
             
             # Try Cerebras first for report generation
-            # logger.info("Attempting report generation with Cerebras...")
-            # try:
-            #     report_files = await generate_csv_report_cerebras(csv_url, query, chat_id, conversation_history)
-            #     if report_files is not None and (report_files.files.csv_files or report_files.files.image_files):
-            #         logger.info(f"Cerebras report generation successful: {len(report_files.files.csv_files)} CSV files, {len(report_files.files.image_files)} image files")
-            #         return {"answer": jsonable_encoder(report_files)}
-            #     else:
-            #         logger.warning("Cerebras report generation returned empty or None result")
-            # except Exception as cerebras_error:
-            #     logger.error(f"Cerebras report generation failed: {str(cerebras_error)}")
+            logger.info("Attempting report generation with Cerebras...")
+            try:
+                report_files = await generate_csv_report_cerebras(csv_url, query, chat_id, conversation_history)
+                if report_files is not None and (report_files.files.csv_files or report_files.files.image_files):
+                    logger.info(f"Cerebras report generation successful: {len(report_files.files.csv_files)} CSV files, {len(report_files.files.image_files)} image files")
+                    return {"answer": jsonable_encoder(report_files)}
+                else:
+                    logger.warning("Cerebras report generation returned empty or None result")
+            except Exception as cerebras_error:
+                logger.error(f"Cerebras report generation failed: {str(cerebras_error)}")
             
-            # # Fallback to Gemini for report generation
-            # logger.info("Falling back to Gemini for report generation...")
-            # try:
-            #     report_files = await generate_csv_report_gemini(csv_url, query, chat_id, conversation_history)
-            #     if report_files is not None and (report_files.files.csv_files or report_files.files.image_files):
-            #         logger.info(f"Gemini report generation successful: {len(report_files.files.csv_files)} CSV files, {len(report_files.files.image_files)} image files")
-            #         return {"answer": jsonable_encoder(report_files)}
-            #     else:
-            #         logger.warning("Gemini report generation returned empty or None result")
-            # except Exception as gemini_error:
-            #     logger.error(f"Gemini report generation failed: {str(gemini_error)}")
+            # Fallback to Gemini for report generation
+            logger.info("Falling back to Gemini for report generation...")
+            try:
+                report_files = await generate_csv_report_gemini(csv_url, query, chat_id, conversation_history)
+                if report_files is not None and (report_files.files.csv_files or report_files.files.image_files):
+                    logger.info(f"Gemini report generation successful: {len(report_files.files.csv_files)} CSV files, {len(report_files.files.image_files)} image files")
+                    return {"answer": jsonable_encoder(report_files)}
+                else:
+                    logger.warning("Gemini report generation returned empty or None result")
+            except Exception as gemini_error:
+                logger.error(f"Gemini report generation failed: {str(gemini_error)}")
             
-            # logger.error("Both Cerebras and Gemini report generation failed")
+            logger.error("Both Cerebras and Gemini report generation failed")
            
-           # Gemini failed, last try Groq Report Generation
-            logger.info("Attempting report generation with Groq as last try...")
+           # Gemini failed, last resort Groq Report Generation
+            logger.info("Attempting report generation with Groq as last resort...")
             try:
                 report_files = await generate_csv_report_groq(csv_url, query, chat_id, conversation_history)
                 if report_files is not None and (report_files.files.csv_files or report_files.files.image_files):