done

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, File, UploadFile, Response, HTTPException
 from fastapi.responses import JSONResponse, FileResponse, StreamingResponse
 from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
+from datetime import datetime
 import io
 from io import BytesIO
 import requests
@@ -287,22 +288,25 @@ async def generate_excel(predictions: List[Prediction]):
         #     sheet.add_image(img, img_cell)
 
 
-    # Save the Excel file to a temporary path
-    excel_file_path = "/predictions_with_images.xlsx"
+    # Create a unique filename based on the current timestamp or index
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    excel_file_path = f"/predictions_with_images_{timestamp}.xlsx"
+    
+    # Save the Excel file to the specified path
     workbook.save(excel_file_path)
 
-     # Upload the Excel file to File.io
-    with open(excel_file_path, 'rb') as file:
-        response = requests.post('https://file.io/', files={'file': file})
-        response_data = response.json()
-
-    # Check if upload was successful
-    if response.status_code == 200:
-        file_url = response_data['link']
-        print(file_url)
-        return JSONResponse(content={"download_link": file_url})
-    else:
-        return JSONResponse(status_code=500, content={"error": "File upload failed"})
+    # Check if the directory exists, if not, create it (to store multiple files)
+    if not os.path.exists("/predictions"):
+        os.makedirs("/predictions")
+
+    # Move the file to a new directory
+    os.rename(excel_file_path, f"/predictions/{os.path.basename(excel_file_path)}")
+    hf_path = "https://huggingface.co/spaces/root-sajjan/whatisit/resolve/main"
+    excel_file_path = hf_path + f"/predictions/{os.path.basename(excel_file_path)}"+"?download=True"
+    
+    return JSONResponse(content={"download_link": excel_file_path})
+    # else:
+    #     return JSONResponse(status_code=500, content={"error": "File upload failed"})
 
     
 '''