Update app.py

app.py

@@ -417,9 +417,16 @@ class CodeGenController:
             self.current_files = scaffold["files"]
             self.current_code = "\n\n".join(f"# {n}\n{c}" for n, c in self.current_files.items())
             self.best_files = dict(self.current_files)
+            # Ensure we have at least requirements.txt
+            if "requirements.txt" not in self.best_files:
+                self.best_files["requirements.txt"] = "# Add your requirements here"
             return True
-        self.current_files = {"main.py": "# New project"}
-        self.current_code = "# New project"
+        # Better defaults if scaffolding fails
+        self.current_files = {
+            "main.py": "# New project\n\ndef main():\n    print('Hello, World!')\n\nif __name__ == '__main__':\n    main()",
+            "requirements.txt": "# Add your requirements here"
+        }
+        self.current_code = "\n\n".join(f"# {n}\n{c}" for n, c in self.current_files.items())
         self.best_files = dict(self.current_files)
         return False
     
@@ -546,10 +553,10 @@ Return the perfected code in META format."""
     def format_output(self, log_msg: str, iteration: int, max_iters: int, state: Optional[Dict] = None):
         progress = f"Iteration {iteration}/{max_iters if max_iters < 999999 else 'INF'}" if iteration <= max_iters else "Complete"
         
-        main = self.best_files.get("main.py", "")
-        test = next((v for k, v in self.best_files.items() if 'test' in k and k.endswith('.py')), "")
-        req = self.best_files.get("requirements.txt", "")
-        readme = self.best_files.get("README.md", "")
+        main = self.best_files.get("main.py", "# Generating code...")
+        test = next((v for k, v in self.best_files.items() if 'test' in k and k.endswith('.py')), "# No tests yet...")
+        req = self.best_files.get("requirements.txt", "# No requirements yet...")
+        readme = self.best_files.get("README.md", "# Generating README...")
         other = {k: v for k, v in self.best_files.items() if k not in [
             "main.py", next((k for k in self.best_files if 'test' in k and k.endswith('.py')), None), 
             "requirements.txt", "README.md"