Refactor project generation response handling to ensure content is created while the temp directory is available and improve error handling for file reading.

Author: Acuspeedster

app/main.py

@@ -639,56 +639,40 @@ async def generate_project_sync(request: ProjectRequest):
             # Try compiling again
             success, output = compiler.build_project(temp_dir)
 
+        # Create the response content while the tempdir is still available
+        all_files_content = ""
+        for f in file_paths:
+            try:
+                file_path = os.path.join(temp_dir, f)
+                if os.path.exists(file_path):
+                    with open(file_path, 'r') as file:
+                        all_files_content += f"[filename: {f}]\n{file.read()}\n\n"
+            except Exception as e:
+                print(f"Error reading file {f}: {e}")
+        
         if success:
-            # Project compiled successfully, try running it
-            run_success, run_output = compiler.run_project(temp_dir)
-            
+            # Project compiled successfully
             status.update({
                 "status": "completed",
                 "message": "Project generated successfully",
-                "build_output": output,
-                "run_output": run_output if run_success else "Failed to run project"
+                "build_output": output
             })
 
-            # Create the response content while the tempdir is still available
-            all_files_content = ""
-            for f in file_paths:
-                try:
-                    file_path = os.path.join(temp_dir, f)
-                    if os.path.exists(file_path):
-                        with open(file_path, 'r') as file:
-                            all_files_content += f"[filename: {f}]\n{file.read()}\n\n"
-                except Exception as e:
-                    print(f"Error reading file {f}: {e}")
-            
             # Add build status
             all_files_content += "\n# Build succeeded\n"
-            
-            # Return the response while still inside the context
-            return PlainTextResponse(content=all_files_content)
         else:
+            # Build failed
             status.update({
                 "status": "failed",
                 "message": "Failed to generate working project",
                 "build_output": output
             })
 
-            # Add a return statement here too, don't let execution continue outside the with block
-            all_files_content = ""
-            for f in file_paths:
-                try:
-                    file_path = os.path.join(temp_dir, f)
-                    if os.path.exists(file_path):
-                        with open(file_path, 'r') as file:
-                            all_files_content += f"[filename: {f}]\n{file.read()}\n\n"
-                except Exception as e:
-                    print(f"Error reading file {f}: {e}")
-            
             # Add build status
             all_files_content += "\n# Build failed\n"
-            
-            save_status(temp_dir, status)
-            return PlainTextResponse(content=all_files_content)
+        
+        save_status(temp_dir, status)
+        return PlainTextResponse(content=all_files_content)
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error generating project: {str(e)}")