updated backend for cookies

evolutiontransformer/api.py

@@ -22,7 +22,10 @@ app = FastAPI()
 
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Allow all origins for now to debug
+    allow_origins=[
+        "http://localhost:5173",
+        "https://tcmmichaelb139-evolutiontransformer.hf.space",
+    ],  # Allow all origins for now to debug
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -53,13 +56,12 @@ def get_session_id(request: Request, response: Response):
     if not session_id:
         session_id = str(uuid.uuid4())
         print(f"Generated new session_id: {session_id}")
-        # Set cookie with appropriate settings
         response.set_cookie(
             key="session_id",
             value=session_id,
-            httponly=True,  # Prevent XSS attacks
-            secure=True,  # Only send over HTTPS in production
-            samesite="lax",  # Allow same-site requests
+            httponly=True,
+            secure=True,
+            samesite="none",
         )
 
     return session_id

frontend/src/App.jsx

@@ -23,15 +23,12 @@ function App() {
 
     const loadModels = async () => {
       try {
-        console.log("Loading models...");
         const taskId = await fetchModels();
-        console.log("Got task ID:", taskId);
 
         if (taskId) {
           checkTaskStatus(
             taskId,
             (result) => {
-              console.log("Models loaded successfully:", result);
               if (result && Array.isArray(result.response)) {
                 setModels(result.response);
               }