added gemini too

controller.py

@@ -178,6 +178,7 @@ async def get_image(request: ImageRequest, authorization: str = Header(None)):
         logger.info("Uploading the chart to supabase...")
         image_public_url = await upload_file_to_supabase(f"{image_file_path}", unique_file_name)
         logger.info("Image uploaded to Supabase and Image URL is... ", {image_public_url})
+        os.remove(image_file_path)
         return {"image_url": image_public_url}
         # return FileResponse(image_file_path, media_type="image/png")
     except Exception as e:
@@ -868,6 +869,7 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
                 logger.info("Uploading the chart to supabase...")
                 image_public_url = await upload_file_to_supabase(f"{langchain_result[0]}", unique_file_name)
                 logger.info("Image uploaded to Supabase and Image URL is... ", {image_public_url})
+                os.remove(langchain_result[0])
                 return {"image_url": image_public_url}
                # return FileResponse(langchain_result[0], media_type="image/png")
                
@@ -890,6 +892,7 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
             logger.info("Uploading the chart to supabase...")
             image_public_url = await upload_file_to_supabase(f"{groq_result}", unique_file_name)
             logger.info("Image uploaded to Supabase and Image URL is... ", {image_public_url})
+            os.remove(groq_result)
             return {"image_url": image_public_url}
             # return FileResponse(groq_result, media_type="image/png")
           
@@ -904,6 +907,7 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
            logger.info("Uploading the chart to supabase...")
            image_public_url = await upload_file_to_supabase(f"{langchain_paths[0]}", unique_file_name)
            logger.info("Image uploaded to Supabase and Image URL is... ", {image_public_url})
+           os.remove(langchain_paths[0])
            return {"image_url": image_public_url}
            # return FileResponse(langchain_paths[0], media_type="image/png")
         else:

gemini_report_generator.py

@@ -301,9 +301,10 @@ async def generate_csv_report(csv_url: str, query: str) -> FileBoxProps:
                 if os.path.exists(csv_path):
                     file_name = os.path.basename(csv_path)
                     try:
+                        unique_file_name = f"{uuid.uuid4()}_{file_name}"
                         public_url = await upload_file_to_supabase(
                             file_path=csv_path,
-                            file_name=file_name
+                            file_name=unique_file_name
                         )
                         csv_files.append(FileProps(
                             fileName=file_name,
@@ -320,9 +321,10 @@ async def generate_csv_report(csv_url: str, query: str) -> FileBoxProps:
                 if os.path.exists(img_path):
                     file_name = os.path.basename(img_path)
                     try:
+                        unique_file_name = f"{uuid.uuid4()}_{file_name}"
                         public_url = await upload_file_to_supabase(
                             file_path=img_path,
-                            file_name=file_name
+                            file_name=unique_file_name
                         )
                         image_files.append(FileProps(
                             fileName=file_name,

orchestrator_functions.py

@@ -590,6 +590,7 @@ async def csv_chart(csv_url: str, query: str):
         unique_name = f'{uuid.uuid4()}.png'
         public_url = await upload_file_to_supabase(image_path, unique_name)
         print(f"Uploaded chart: {public_url}")
+        os.remove(image_path) # Remove the local image file after upload
         return {"image_url": public_url}
 
     try: