Update app.py

app.py

@@ -1,14 +1,24 @@
 from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
 import base64
 import os
 
 app = FastAPI()
 
-# Load LoRA weights on startup
+# Middleware to handle CORS (optional, but useful if Testing4 calls this API)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Adjust as needed for security
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Load LoRA weights globally
 lora_weights = None
 
 @app.on_event("startup")
-def load_lora_weights():
+async def startup_event():
     global lora_weights
     lora_path = "./lora_file.pth"
     if os.path.exists(lora_path):
@@ -17,6 +27,7 @@ def load_lora_weights():
             lora_weights = base64.b64encode(f.read()).decode("utf-8")
             print("LoRA weights loaded and preprocessed successfully.")
     else:
+        print("LoRA file not found during startup.")
         raise HTTPException(status_code=500, detail="LoRA file not found.")
 
 @app.post("/modify-prompt")