added gemini too

gemini_report_generator.py

@@ -286,55 +286,68 @@ def gemini_llm_chat(csv_url: str, query: str) -> Dict[str, Any]:
         }
         
 
-async def generate_csv_report(csv_url: str, query: str) -> FileBoxProps:
+async def generate_report(csv_url: str, query: str) -> FileBoxProps:
     try:
         result = gemini_llm_chat(csv_url, query)
-        logger.info(f"Report generated successfully: {result}")
+        logger.info(f"Raw result from gemini_llm_chat: {result}")
         
-        # Prepare lists for files
         csv_files = []
         image_files = []
         
-        # Assuming result contains 'csv_files' and 'image_files' keys
-        if 'csv_files' in result:
-            for csv_file in result['csv_files']:
-                # Save CSV to Supabase
-                public_url = await upload_file_to_supabase(
-                    file_path=csv_file['local_path'],
-                    file_name=csv_file['name']
-                )
-                csv_files.append(FileProps(
-                    fileName=csv_file['name'],
-                    filePath=public_url,
-                    fileType="csv"
-                ))
-                # Clean up local file
-                os.remove(csv_file['local_path'])
-        
-        if 'image_files' in result:
-            for img_file in result['image_files']:
-                # Save image to Supabase
-                public_url = await upload_file_to_supabase(
-                    file_path=img_file['local_path'],
-                    file_name=img_file['name']
+        # Check if we got the expected response structure
+        if isinstance(result, dict) and 'csv_files' in result and 'image_files' in result:
+            # Process CSV files
+            for csv_path in result['csv_files']:
+                if os.path.exists(csv_path):
+                    file_name = os.path.basename(csv_path)
+                    try:
+                        public_url = await upload_file_to_supabase(
+                            file_path=csv_path,
+                            file_name=file_name
+                        )
+                        csv_files.append(FileProps(
+                            fileName=file_name,
+                            filePath=public_url,
+                            fileType="csv"
+                        ))
+                        os.remove(csv_path)  # Clean up
+                    except Exception as upload_error:
+                        logger.error(f"Failed to upload CSV {file_name}: {str(upload_error)}")
+                        continue
+            
+            # Process image files
+            for img_path in result['image_files']:
+                if os.path.exists(img_path):
+                    file_name = os.path.basename(img_path)
+                    try:
+                        public_url = await upload_file_to_supabase(
+                            file_path=img_path,
+                            file_name=file_name
+                        )
+                        image_files.append(FileProps(
+                            fileName=file_name,
+                            filePath=public_url,
+                            fileType="image"
+                        ))
+                        os.remove(img_path)  # Clean up
+                    except Exception as upload_error:
+                        logger.error(f"Failed to upload image {file_name}: {str(upload_error)}")
+                        continue
+            
+            return FileBoxProps(
+                files=Files(
+                    csv_files=csv_files,
+                    image_files=image_files
                 )
-                image_files.append(FileProps(
-                    fileName=img_file['name'],
-                    filePath=public_url,
-                    fileType="image"
-                ))
-                # Clean up local file
-                os.remove(img_file['local_path'])
-        
-        return FileBoxProps(
-            files=Files(
-                csv_files=csv_files,
-                image_files=image_files
             )
-        )
-        
+        else:
+            raise ValueError("Unexpected response format from gemini_llm_chat")
+            
     except Exception as e:
         logger.error(f"Report generation failed: {str(e)}")
+        # 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=[],