Lets go

controller.py

@@ -17,7 +17,7 @@ from dotenv import load_dotenv
 from pydantic import BaseModel
 from csv_service import clean_data, extract_chart_filenames
 from urllib.parse import unquote
-from csv_service import generate_csv_data, get_csv_basic_info, get_image_by_file_name
+import csv_service
 from langchain_groq import ChatGroq
 import pandas as pd
 from langchain_experimental.tools import PythonAstREPLTool
@@ -43,17 +43,6 @@ logger = logging.getLogger(__name__)
 max_cpus = os.cpu_count()
 logger.info(f"Max CPUs: {max_cpus}")
 
-
-# Thread-safe configuration for chart endpoints
-current_langchain_chart_key_index = 0
-current_langchain_chart_lock = threading.Lock()
-
-# running loop for asyncio
-loop = asyncio.get_running_loop()
-
-# Use a process pool to run CPU-bound chart generation
-process_executor = ProcessPoolExecutor(max_workers=10)
-
 # Ensure the cache directory exists
 os.makedirs("/app/cache", exist_ok=True)
 
@@ -110,13 +99,7 @@ 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 = get_csv_basic_info(decoded_url)          
-       
-        # Run the synchronous function in a thread pool executor
-        csv_data = await loop.run_in_executor(
-            process_executor, get_csv_basic_info, decoded_url
-        )
-       
+        csv_data = csv_service.get_csv_basic_info(decoded_url)    
         logger.info(f"CSV data fetched successfully: {csv_data}")
         return {"data": csv_data}
     except Exception as e:
@@ -153,11 +136,7 @@ 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 = generate_csv_data(decoded_url)     
-        # Run the synchronous function in a thread pool executor
-        csv_data = await loop.run_in_executor(
-            process_executor, generate_csv_data, decoded_url
-        )   
+        csv_data = csv_service.generate_csv_data(decoded_url)        
         return csv_data
     except Exception as e:
         logger.error(f"Error while fetching CSV data: {e}")
@@ -606,8 +585,13 @@ def groq_chart(csv_url: str, question: str):
 # Global locks for key rotation (chart endpoints)
 # current_groq_chart_key_index = 0
 # current_groq_chart_lock = threading.Lock()
+current_langchain_chart_key_index = 0
+current_langchain_chart_lock = threading.Lock()
 
 
+# Use a process pool to run CPU-bound chart generation
+process_executor = ProcessPoolExecutor(max_workers=10)
+
 # --- GROQ-BASED CHART GENERATION ---
 # def groq_chart(csv_url: str, question: str):
 #     """
@@ -769,7 +753,8 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
     try:
         query = request.get("query", "")
         csv_url = unquote(request.get("csv_url", ""))
-
+        
+        loop = asyncio.get_running_loop()
         # First, try the langchain-based method if the question qualifies
         if if_initial_chart_question(query):
             langchain_result = await loop.run_in_executor(