added csv-code executor

controller.py

@@ -7,14 +7,14 @@ import threading
 import uuid
 from fastapi import FastAPI, HTTPException, Header
 from fastapi.encoders import jsonable_encoder
-from typing import Dict, List, Optional
+from typing import Dict, List, Literal, Optional
 from fastapi.responses import FileResponse
 import numpy as np
 import pandas as pd
 from pandasai import SmartDataframe
 from langchain_groq.chat_models import ChatGroq
 from dotenv import load_dotenv
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 from csv_service import clean_data, extract_chart_filenames, generate_csv_data, get_csv_basic_info
 from urllib.parse import unquote
 from langchain_groq import ChatGroq
@@ -28,7 +28,7 @@ import seaborn as sns
 from gemini_report_generator import generate_csv_report
 from intitial_q_handler import if_initial_chart_question, if_initial_chat_question
 from orchestrator_agent import csv_orchestrator_chat
-from python_code_executor_service import PythonExecutor
+from python_code_executor_service import CsvChatResult, PythonExecutor
 from supabase_service import upload_file_to_supabase
 from cerebras_csv_agent import query_csv_agent
 from util_service import _prompt_generator, process_answer
@@ -206,66 +206,31 @@ async def get_csv_data(request: CsvUrlRequest):
         raise HTTPException(status_code=400, detail=f"Failed to retrieve CSV data: {str(e)}")
 
 # EXECUTE THE PYTHON CODE 
-class CodeResponse(BaseModel):
-    language: str
-    code: str
-
-class AnalysisOperation(BaseModel):
-    code: CodeResponse
-    resultVar: str
-
-class ChartSpecification(BaseModel):
-    imageDescription: str
-    code: Optional[str] = None
-
-class CsvChatResult(BaseModel):
-    casualResponse: str
-    analysisOperations: List[AnalysisOperation]
-    charts: Optional[List[ChartSpecification]] = None
-
 class ExecutionRequest(BaseModel):
-    result: CsvChatResult
-    csvUrl: str
-    chatId: str
-    
-class CsvChatResultPython(BaseModel):
-    """Structured response for CSV-related AI interactions"""
-    casual_response: str
-    analysis_operations: List[AnalysisOperation]
-    charts: Optional[List[ChartSpecification]] = None
+    chat_id: str = Field(..., alias="chat_id")
+    csv_url: str = Field(..., alias="csv_url")
+    response_type: Literal["casual", "data_analysis", "visualization", "mixed"]
+    codeExecutionPayload: CsvChatResult
 
 @app.post("/api/code_execution_csv")
 async def code_execution_csv(
     request: ExecutionRequest,
     authorization: Optional[str] = Header(None)
 ):
-    # Token validation
     expected_token = os.environ.get("AUTH_TOKEN")
     if not authorization or not expected_token or authorization.replace("Bearer ", "") != expected_token:
         raise HTTPException(status_code=401, detail="Unauthorized")
 
     try:
-
-        # Initialize executor with DataFrame
-        csv_url = request.csvUrl
-        decoded_url = unquote(csv_url)
+        decoded_url = unquote(request.csv_url)
         df = clean_data(decoded_url)
         executor = PythonExecutor(df)
-        result_body = CsvChatResultPython(
-            casual_response=request.result.casualResponse,
-            analysis_operations=request.result.analysisOperations,
-            charts=request.result.charts
-        )
 
-        formatted_output = await executor.process_response(result_body, request.chatId)
-        logger.info("Formatted Output:", formatted_output)
+        formatted_output = await executor.process_response(request.codeExecutionPayload, request.chat_id)
         return {"answer": formatted_output}
 
     except Exception as e:
-        return {
-            "error": "Failed to execute request",
-            "message": str(e)
-        }
+        return {"error": "Failed to execute request", "message": str(e)}
 
 
 # CHAT CODING STARTS FROM HERE

python_code_executor_service.py

@@ -42,6 +42,7 @@ class AnalysisOperation(BaseModel):
 
 class CsvChatResult(BaseModel):
     """Structured response for CSV-related AI interactions"""
+    response_type: Literal["casual", "data_analysis", "visualization", "mixed"]
     casual_response: str
     analysis_operations: List[AnalysisOperation]
     charts: Optional[List[ChartSpecification]] = None