added gemini too

controller.py

@@ -351,7 +351,7 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
         if generate_report is True:
             report_files = await generate_csv_report(csv_url, query)
             if report_files is not None:
-                return {"orchestrator_response": jsonable_encoder(report_files)}
+                return {"answer": jsonable_encoder(report_files)}
 
         if if_initial_chat_question(query):
             answer = await asyncio.to_thread(

gemini_report_generator.py

@@ -350,12 +350,13 @@ async def generate_csv_report(csv_url: str, query: str) -> FileBoxProps:
         # Return empty response but log the files we found
         if 'csv_files' in locals() and 'image_files' in locals():
             logger.info(f"Files that were generated but not processed: CSV: {result.get('csv_files', [])}, Images: {result.get('image_files', [])}")
-        return FileBoxProps(
-            files=Files(
-                csv_files=[],
-                image_files=[]
-            )
-        )
+        # return FileBoxProps(
+        #     files=Files(
+        #         csv_files=[],
+        #         image_files=[]
+        #     )
+        # )
+        return None
 
 
 # if __name__ == "__main__":