added gemini too

controller.py

@@ -7,7 +7,7 @@ import threading
 import uuid
 from fastapi import FastAPI, HTTPException, Header
 from fastapi.encoders import jsonable_encoder
-from typing import Dict
+from typing import Dict, List
 from fastapi.responses import FileResponse
 import numpy as np
 import pandas as pd
@@ -75,6 +75,48 @@ class CsvUrlRequest(BaseModel):
     
 class ImageRequest(BaseModel):
     image_path: str
+    
+class FileProps(BaseModel):
+    fileName: str
+    filePath: str
+    fileType: str  # 'csv' | 'image'
+
+class Files(BaseModel):
+    csv_files: List[FileProps]
+    image_files: List[FileProps]
+
+class FileBoxProps(BaseModel):
+    files: Files
+    
+dummy_response = FileBoxProps(
+            files=Files(
+                csv_files=[
+                    FileProps(
+                        fileName="sales_data.csv",
+                        filePath="/downloads/sales_data.csv",
+                        fileType="csv"
+                    ),
+                    FileProps(
+                        fileName="customer_data.csv",
+                        filePath="/downloads/customer_data.csv",
+                        fileType="csv"
+                    )
+                ],
+                image_files=[
+                    FileProps(
+                        fileName="chart.png",
+                        filePath="/downloads/chart.png",
+                        fileType="image"
+                    ),
+                    FileProps(
+                        fileName="graph.jpg",
+                        filePath="/downloads/graph.jpg",
+                        fileType="image"
+                    )
+                ]
+            )
+        )
+
 
 # Thread-safe key management for groq_chat
 current_groq_key_index = 0
@@ -305,7 +347,7 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
         generate_report = request.get("generate_report")
         
         if generate_report is True:
-            return {"answer": "Generate report is True"}
+            return {"answer": jsonable_encoder(dummy_response)}
 
         if if_initial_chat_question(query):
             answer = await asyncio.to_thread(
@@ -806,7 +848,7 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
         conversation_history = request.get("conversation_history", [])
         generate_report = request.get("generate_report", False)
         if generate_report is True:
-           return {"orchestrator_response": "Generate report is True"}
+           return {"orchestrator_response": jsonable_encoder(dummy_response)}
 
         loop = asyncio.get_running_loop()
         # First, try the langchain-based method if the question qualifies