revert route

Author: pengyu

frontend/src/app/api/runProject/route.ts

@@ -517,12 +517,10 @@ export async function GET(req: Request) {
       existingContainer.containerId
     );
     if (isRunning) {
-      logger.info(`Using existing container with port: ${existingContainer.port}`);
       return NextResponse.json({
         message: 'Docker container already running',
         domain: existingContainer.domain,
         containerId: existingContainer.containerId,
-        port: existingContainer.port,
       });
     } else {
       // Remove non-running container from state
@@ -549,14 +547,12 @@ export async function GET(req: Request) {
   processingRequests.add(projectPath);
 
   try {
-    const { domain, containerId, port } = await runDockerContainer(projectPath);
-    logger.info(`Successfully started container on port: ${port}`);
+    const { domain, containerId } = await runDockerContainer(projectPath);
 
     return NextResponse.json({
       message: 'Docker container started',
       domain,
       containerId,
-      port,
     });
   } catch (error: any) {
     logger.error(`Failed to start Docker container:`, error);
@@ -567,4 +563,4 @@ export async function GET(req: Request) {
   } finally {
     processingRequests.delete(projectPath);
   }
-}
+}

frontend/src/app/api/screenshot/route.ts

@@ -22,7 +22,6 @@ export async function GET(req: Request) {
   const { searchParams } = new URL(req.url);
   const url = searchParams.get('url');
   let page = null;
-  const MAX_RETRIES = 3;
 
   if (!url) {
     return NextResponse.json(
@@ -31,15 +30,12 @@ export async function GET(req: Request) {
     );
   }
 
-  for (let attempt = 0; attempt < MAX_RETRIES; attempt++) {
-    try {
-      logger.info(`Screenshot attempt ${attempt + 1} for ${url}`);
-      
-      // Get browser instance
-      const browser = await getBrowser();
+  try {
+    // Get browser instance
+    const browser = await getBrowser();
 
-      // Create a new page
-      page = await browser.newPage();
+    // Create a new page
+    page = await browser.newPage();
 
     // Set viewport to a reasonable size
     await page.setViewport({
@@ -77,66 +73,51 @@ export async function GET(req: Request) {
       fullPage: true,
     });
 
-      // Always close the page when done
-      if (page) {
-        await page.close();
-      }
+    // Always close the page when done
+    if (page) {
+      await page.close();
+    }
 
-      // Return the screenshot as a PNG image
-      return new Response(screenshot, {
-        headers: {
-          'Content-Type': 'image/png',
-          'Cache-Control': 's-maxage=3600',
-        },
-      });
-    } catch (error: any) {
-      logger.error(`Screenshot error on attempt ${attempt + 1}:`, error);
-
-      // Ensure page is closed even if an error occurs
-      if (page) {
-        try {
-          await page.close();
-        } catch (closeError) {
-          logger.error('Error closing page:', closeError);
-        }
-      }
+    // Return the screenshot as a PNG image
+    return new Response(screenshot, {
+      headers: {
+        'Content-Type': 'image/png',
+        'Cache-Control': 's-maxage=3600',
+      },
+    });
+  } catch (error: any) {
+    logger.error('Screenshot error:', error);
 
-      // If browser seems to be in a bad state, recreate it
-      if (
-        error.message.includes('Target closed') ||
-        error.message.includes('Protocol error') ||
-        error.message.includes('Target.createTarget')
-      ) {
-        try {
-          if (browserInstance) {
-            await browserInstance.close();
-            browserInstance = null;
-          }
-        } catch (closeBrowserError) {
-          logger.error('Error closing browser:', closeBrowserError);
-        }
+    // Ensure page is closed even if an error occurs
+    if (page) {
+      try {
+        await page.close();
+      } catch (closeError) {
+        logger.error('Error closing page:', closeError);
       }
+    }
 
-      // 如果这不是最后一次尝试，则继续
-      if (attempt < MAX_RETRIES - 1) {
-        // 等待一会儿再重试
-        await new Promise(resolve => setTimeout(resolve, 3000));
-        continue;
+    // If browser seems to be in a bad state, recreate it
+    if (
+      error.message.includes('Target closed') ||
+      error.message.includes('Protocol error') ||
+      error.message.includes('Target.createTarget')
+    ) {
+      try {
+        if (browserInstance) {
+          await browserInstance.close();
+          browserInstance = null;
+        }
+      } catch (closeBrowserError) {
+        logger.error('Error closing browser:', closeBrowserError);
       }
-
-      // 最后一次尝试失败
-      return NextResponse.json(
-        { error: error.message || 'Failed to capture screenshot after multiple attempts' },
-        { status: 500 }
-      );
     }
-  }
 
-  // 如果重试都失败
-  return NextResponse.json(
-    { error: 'Failed to capture screenshot after exhausting all retries' },
-    { status: 500 }
-  );
+    return NextResponse.json(
+      { error: error.message || 'Failed to capture screenshot' },
+      { status: 500 }
+    );
+  }
 }
 
 // Handle process termination to close browser
@@ -147,4 +128,4 @@ process.on('SIGINT', async () => {
     browserInstance = null;
   }
   process.exit(0);
-});
+});

frontend/src/components/chat/code-engine/project-context.tsx

@@ -603,33 +603,36 @@ export function ProjectProvider({ children }: { children: ReactNode }) {
 
   const takeProjectScreenshot = useCallback(
     async (projectId: string, url: string): Promise<void> => {
-      // Check if this screenshot operation is already in progress
       const operationKey = `screenshot_${projectId}`;
       if (pendingOperations.current.get(operationKey)) {
+        logger.debug(`[screenshot] Project ${projectId} is already being processed`);
         return;
       }
-
+  
       pendingOperations.current.set(operationKey, true);
-
+      logger.debug(`[screenshot] Start for Project ${projectId}, URL: ${url}`);
+  
       try {
-        // Check if the URL is accessible
+        logger.debug(`[screenshot] Checking accessibility for ${url}`);
         const isUrlAccessible = await checkUrlStatus(url);
         if (!isUrlAccessible) {
-          logger.warn(`URL ${url} is not accessible after multiple retries`);
+          logger.warn(`[screenshot] URL ${url} is not accessible after retries`);
           return;
         }
-
-        // Add a cache buster to avoid previous screenshot caching
+  
         const screenshotUrl = `/api/screenshot?url=${encodeURIComponent(url)}&t=${Date.now()}`;
+        logger.debug(`[screenshot] Sending request to ${screenshotUrl}`);
         const screenshotResponse = await fetch(screenshotUrl);
-
+  
         if (!screenshotResponse.ok) {
           throw new Error(
-            `Failed to capture screenshot: ${screenshotResponse.status} ${screenshotResponse.statusText}`
+            `[screenshot] Failed to capture: ${screenshotResponse.status} ${screenshotResponse.statusText}`
           );
         }
-
+  
         const arrayBuffer = await screenshotResponse.arrayBuffer();
+        logger.debug(`[screenshot] Screenshot captured for Project ${projectId}, uploading...`);
+  
         const blob = new Blob([arrayBuffer], { type: 'image/png' });
         const file = new File([blob], 'screenshot.png', { type: 'image/png' });
 
@@ -642,13 +645,15 @@ export function ProjectProvider({ children }: { children: ReactNode }) {
           },
         });
       } catch (error) {
-        logger.error('Error taking screenshot:', error);
+        logger.error(`[screenshot] Error for Project ${projectId}:`, error);
       } finally {
+        logger.debug(`[screenshot] Finished process for Project ${projectId}`);
         pendingOperations.current.delete(operationKey);
       }
     },
     [updateProjectPhotoMutation]
   );
+  
 
   const getWebUrl = useCallback(
     async (