Lets go

controller.py

@@ -15,9 +15,8 @@ from pandasai import SmartDataframe
 from langchain_groq.chat_models import ChatGroq
 from dotenv import load_dotenv
 from pydantic import BaseModel
-from csv_service import clean_data, extract_chart_filenames
+from csv_service import clean_data, extract_chart_filenames, generate_csv_data, get_csv_basic_info
 from urllib.parse import unquote
-import csv_service
 from langchain_groq import ChatGroq
 import pandas as pd
 from langchain_experimental.tools import PythonAstREPLTool
@@ -99,7 +98,12 @@ async def basic_csv_data(request: CsvUrlRequest):
     try:
         decoded_url = unquote(request.csv_url)
         logger.info(f"Fetching CSV data from URL: {decoded_url}")
-        csv_data = csv_service.get_csv_basic_info(decoded_url)    
+        # csv_data = await get_csv_basic_info(decoded_url)    
+        # Run the synchronous function in a thread pool executor
+        loop = asyncio.get_running_loop()
+        csv_data = await loop.run_in_executor(
+            process_executor, get_csv_basic_info, decoded_url
+        )
         logger.info(f"CSV data fetched successfully: {csv_data}")
         return {"data": csv_data}
     except Exception as e:
@@ -136,7 +140,11 @@ async def get_csv_data(request: CsvUrlRequest):
     try:
         decoded_url = unquote(request.csv_url)
         logger.info(f"Fetching CSV data from URL: {decoded_url}")
-        csv_data = csv_service.generate_csv_data(decoded_url)        
+        # csv_data = await generate_csv_data(decoded_url)     
+        loop = asyncio.get_running_loop()
+        csv_data = await loop.run_in_executor(
+            process_executor, generate_csv_data, decoded_url
+        )   
         return csv_data
     except Exception as e:
         logger.error(f"Error while fetching CSV data: {e}")