Update main.py

main.py

@@ -112,15 +112,22 @@ def inference(img, version, scale):
 
 
 
+import time
+
 @app.post("/upload/")
 async def upload_image(file: UploadFile = File(...), version: str = Form(...), scale: int = Form(...)):
     try:
+        # Create a unique filename with timestamp
+        timestamp = str(int(time.time()))
+        file_extension = os.path.splitext(file.filename)[1]
+        filename = f"uploaded_image_{timestamp}{file_extension}"
+
         # Save the uploaded file
-        with open(f"uploaded_image{os.path.splitext(file.filename)[1]}", "wb") as buffer:
+        with open(filename, "wb") as buffer:
             shutil.copyfileobj(file.file, buffer)
 
         # Perform image enhancement
-        enhanced_image, save_path = inference(f"uploaded_image{os.path.splitext(file.filename)[1]}", version, scale)
+        enhanced_image, save_path = inference(filename, version, scale)
 
         # Return the enhanced image
         if enhanced_image is not None: