Update app.py

app.py

@@ -67,7 +67,7 @@ def process_image(image, project_name):
         if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
             return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
 
-        # Ensure the 'public_uploads' directory exists
+        # Ensure the 'public_uploads' directory exists before attempting to save
         upload_dir = "public_uploads"
         os.makedirs(upload_dir, exist_ok=True)
 
@@ -123,7 +123,11 @@ def process_image(image, project_name):
 app = FastAPI()
 
 # Ensure the 'public_uploads' directory is created before serving
-app.mount("/file/public_uploads", StaticFiles(directory="public_uploads"), name="public_uploads")
+upload_dir = "public_uploads"
+os.makedirs(upload_dir, exist_ok=True)
+
+# Mount the 'public_uploads' directory to serve static files
+app.mount("/file/public_uploads", StaticFiles(directory=upload_dir), name="public_uploads")
 
 # Gradio UI
 with gr.Blocks(css=".gradio-container {background-color: #f0f4f8; font-family: Arial;} .title {color: #2c3e50; font-size: 24px; text-align: center;}") as demo: