Report generator

orchestrator_functions.py

@@ -385,78 +385,78 @@ async def csv_chart(csv_url: str, query: str):
     
     
     
-# async def csv_chat(csv_url: str, query: str):
-#     try:
-#         updated_query = f"{query} and Do not show any charts or graphs."
+async def csv_chat(csv_url: str, query: str):
+    try:
+        updated_query = f"{query} and Do not show any charts or graphs."
         
-#         # Process with langchain_chat first
-#         try:
-#             lang_answer = await asyncio.to_thread(
-#                 langchain_csv_chat, csv_url, query, False
-#             )
-#             if lang_answer is not None:
-#                 return {"answer": jsonable_encoder(lang_answer)}
-#             raise Exception("Langchain failed to process")
-#         except Exception as langchain_error:
-#             print(f"Langchain error, falling back to Groq: {str(langchain_error)}")
+        # Process with langchain_chat first
+        try:
+            lang_answer = await asyncio.to_thread(
+                langchain_csv_chat, csv_url, query, False
+            )
+            if lang_answer is not None:
+                return {"answer": jsonable_encoder(lang_answer)}
+            raise Exception("Langchain failed to process")
+        except Exception as langchain_error:
+            print(f"Langchain error, falling back to Groq: {str(langchain_error)}")
             
-#             # Process with groq_chat if langchain fails
-#             try:
-#                 groq_answer = await asyncio.to_thread(groq_chat, csv_url, updated_query)
-#                 print("groq_answer:", groq_answer)
+            # Process with groq_chat if langchain fails
+            try:
+                groq_answer = await asyncio.to_thread(groq_chat, csv_url, updated_query)
+                print("groq_answer:", groq_answer)
                 
-#                 if process_answer(groq_answer) == "Empty response received.":
-#                     return {"answer": "Sorry, I couldn't find relevant data..."}
+                if process_answer(groq_answer) == "Empty response received.":
+                    return {"answer": "Sorry, I couldn't find relevant data..."}
                 
-#                 if process_answer(groq_answer) or groq_answer is None:
-#                     return {"answer": "error"}
+                if process_answer(groq_answer) or groq_answer is None:
+                    return {"answer": "error"}
                 
-#                 return {"answer": jsonable_encoder(groq_answer)}
-#             except Exception as groq_error:
-#                 print(f"Groq processing error: {str(groq_error)}")
-#                 return {"answer": "error"}
+                return {"answer": jsonable_encoder(groq_answer)}
+            except Exception as groq_error:
+                print(f"Groq processing error: {str(groq_error)}")
+                return {"answer": "error"}
                 
-#     except Exception as e:
-#         print(f"Error processing request: {str(e)}")
-#         return {"answer": "error"}
+    except Exception as e:
+        print(f"Error processing request: {str(e)}")
+        return {"answer": "error"}
 
 
 
 
 
 
-async def csv_chat(csv_url: str, query: str):
-    try:
-        updated_query = f"{query} and Do not show any charts or graphs."
+# async def csv_chat(csv_url: str, query: str):
+#     try:
+#         updated_query = f"{query} and Do not show any charts or graphs."
         
-        # Process with Groq first
-        try:
-            groq_answer = await asyncio.to_thread(groq_chat, csv_url, updated_query)
-            print("groq_answer:", groq_answer)
+#         # Process with Groq first
+#         try:
+#             groq_answer = await asyncio.to_thread(groq_chat, csv_url, updated_query)
+#             print("groq_answer:", groq_answer)
             
-            if process_answer(groq_answer) == "Empty response received." or groq_answer == None:
-                return {"answer": "Sorry, I couldn't find relevant data..."}
+#             if process_answer(groq_answer) == "Empty response received." or groq_answer == None:
+#                 return {"answer": "Sorry, I couldn't find relevant data..."}
             
-            if process_answer(groq_answer) or groq_answer == None:
-                raise Exception("Groq response not usable, falling back to LangChain")
+#             if process_answer(groq_answer) or groq_answer == None:
+#                 raise Exception("Groq response not usable, falling back to LangChain")
             
-            return {"answer": jsonable_encoder(groq_answer)}
+#             return {"answer": jsonable_encoder(groq_answer)}
             
-        except Exception as groq_error:
-            print(f"Groq error, falling back to LangChain: {str(groq_error)}")
+#         except Exception as groq_error:
+#             print(f"Groq error, falling back to LangChain: {str(groq_error)}")
             
-            # Process with LangChain if Groq fails
-            try:
-                lang_answer = await asyncio.to_thread(
-                    langchain_csv_chat, csv_url, query, False
-                )
-                if not process_answer(lang_answer):
-                    return {"answer": jsonable_encoder(lang_answer)}
-                return {"answer": "Sorry, I couldn't find relevant data..."}
-            except Exception as langchain_error:
-                print(f"LangChain processing error: {str(langchain_error)}")
-                return {"answer": "error"}
+#             # Process with LangChain if Groq fails
+#             try:
+#                 lang_answer = await asyncio.to_thread(
+#                     langchain_csv_chat, csv_url, query, False
+#                 )
+#                 if not process_answer(lang_answer):
+#                     return {"answer": jsonable_encoder(lang_answer)}
+#                 return {"answer": "Sorry, I couldn't find relevant data..."}
+#             except Exception as langchain_error:
+#                 print(f"LangChain processing error: {str(langchain_error)}")
+#                 return {"answer": "error"}
                 
-    except Exception as e:
-        print(f"Error processing request: {str(e)}")
-        return {"answer": "error"}
+#     except Exception as e:
+#         print(f"Error processing request: {str(e)}")
+#         return {"answer": "error"}