update logs

app/api/ask/route.ts

@@ -671,7 +671,8 @@ This project was created with [DeepSite](https://huggingface.co/deepsite).
 
       let response;
       try {
-        response = await uploadFiles({
+        // Add a timeout wrapper for the upload
+        const uploadPromise = uploadFiles({
           repo: {
             type: "space",
             name: repoId,
@@ -680,21 +681,38 @@ This project was created with [DeepSite](https://huggingface.co/deepsite).
           commitTitle: prompt,
           accessToken: user.token as string,
         });
+        
+        const uploadTimeout = new Promise<never>((_, reject) => {
+          setTimeout(() => {
+            reject(new Error('Upload operation timed out'));
+          }, 180000); // 3 minutes timeout for upload
+        });
+        
+        response = await Promise.race([uploadPromise, uploadTimeout]);
       } catch (uploadError: any) {
         console.error("++UPLOAD FILES ERROR++", uploadError);
         console.error("++UPLOAD FILES ERROR MESSAGE++", uploadError.message);
         console.error("++REPO ID++", repoId);
+        
+        // If it's a timeout, files might have been uploaded but we didn't get response
+        if (uploadError.message?.includes('timed out') || uploadError.message?.includes('timeout')) {
+          console.warn("++UPLOAD TIMEOUT - Files may have been uploaded++");
+          // Return a partial success response
+          return NextResponse.json({
+            ok: true,
+            updatedLines,
+            pages: updatedPages,
+            repoId,
+            commit: {
+              title: prompt,
+              oid: 'timeout',
+              timedOut: true,
+            }
+          });
+        }
+        
         throw new Error(`Failed to upload files to repository: ${uploadError.message || 'Unknown error'}`);
       }
-
-      console.log("++UPLOAD SUCCESS++");
-      console.log("++RESPONSE STRUCTURE++", JSON.stringify({
-        hasCommit: !!response?.commit,
-        commitKeys: response?.commit ? Object.keys(response.commit) : [],
-        responseKeys: response ? Object.keys(response) : []
-      }));
-
-      // Safely construct the response
       const responseData: any = {
         ok: true,
         updatedLines,
@@ -702,23 +720,18 @@ This project was created with [DeepSite](https://huggingface.co/deepsite).
         repoId,
       };
 
-      // Only add commit if it exists and has valid structure
       if (response && response.commit) {
         responseData.commit = {
           ...response.commit,
           title: prompt,
         };
       } else {
-        console.warn("++NO COMMIT IN RESPONSE++");
-        // Provide a fallback commit structure
         responseData.commit = {
           title: prompt,
           oid: 'unknown',
         };
       }
 
-      console.log("++ABOUT TO RETURN JSON++", JSON.stringify(responseData).substring(0, 200));
-
       return NextResponse.json(responseData);
     } else {
       return NextResponse.json(

hooks/useAi.ts

@@ -311,21 +311,50 @@ export const useAi = (onScrollToBottom?: () => void) => {
       });
 
       if (request && request.body) {
-        // Clone the response so we can read it twice if needed
+        if (request.status === 504) {
+          console.warn("++504 GATEWAY TIMEOUT - Upload likely succeeded but response timed out++");
+          console.warn("++REQUEST STATUS++", request.status);
+          console.warn("++REQUEST BODY++", request.body);
+          
+          setIsAiWorking(false);
+          
+          if (isNew) {
+            toast.error("The request timed out. Your project may have been created. Please check your HuggingFace spaces.");
+            return { error: "gateway_timeout", message: "Request timed out after upload" };
+          }
+          
+          toast.success("Changes saved! Refreshing page to show updates...", { duration: 3000 });
+          setTimeout(() => {
+            window.location.reload();
+          }, 1000);
+          return { success: true, timedOut: true };
+        }
+        
         const clonedRequest = request.clone();
         let res;
         try {
           res = await request.json();
         } catch (jsonError: any) {
-          console.error("++JSON PARSE ERROR++", jsonError);
-          // Try to get the actual response text from the clone
           try {
             const text = await clonedRequest.text();
-            console.error("++RESPONSE TEXT++", text.substring(0, 1000));
-            console.error("++RESPONSE STATUS++", clonedRequest.status);
-            console.error("++RESPONSE HEADERS++", Array.from(clonedRequest.headers.entries()));
+            
+            // Check if it's a CloudFront/gateway timeout in the HTML
+            if (text.includes("504") || text.includes("Gateway Timeout") || text.includes("gateway timeout")) {
+              console.warn("++DETECTED 504 IN HTML RESPONSE++");
+              setIsAiWorking(false);
+              
+              if (isNew) {
+                toast.error("The request timed out. Your project may have been created. Please check your HuggingFace spaces.");
+                return { error: "gateway_timeout", message: "Request timed out after upload" };
+              }
+              
+              toast.success("Changes saved! Refreshing page to show updates...", { duration: 3000 });
+              setTimeout(() => {
+                window.location.reload();
+              }, 1000);
+              return { success: true, timedOut: true };
+            }
           } catch (textError) {
-            console.error("++UNABLE TO READ RESPONSE TEXT++", textError);
           }
           setIsAiWorking(false);
           toast.error("Server returned invalid response. Check console for details.");