Fixed async handling and imports, improved progress phases

app.py

@@ -1,11 +1,5 @@
-import gradio as gr
-import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-from langchain_community.llms import HuggingFacePipeline
-from langchain.agents import initialize_agent, Tool
-from langchain.prompts import PromptTemplate
+import os
 import json
-import subprocess
 import logging
 import asyncio
 import time
@@ -15,13 +9,17 @@ from typing import Dict, List, Optional, Set, Union, Any
 from dataclasses import dataclass, field
 from pathlib import Path
 import hashlib
-import os
-import json
-import networkx as nx
-from dataclasses import asdict, field
-import shutil
 import tempfile
-from zipfile import ZipFile
+import shutil
+import gradio as gr
+import networkx as nx
+from langchain.prompts import PromptTemplate
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from langchain_community.llms import HuggingFacePipeline
+from langchain.agents import initialize_agent, Tool
+import subprocess
+import asyncio
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -1062,7 +1060,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as ui:
                         show_copy_button=True
                     )
     
-    def stream_output(requirements, progress=gr.Progress()):
+    async def stream_output(requirements, progress=gr.Progress()):
         """Stream the output during app generation."""
         try:
             # Initialize
@@ -1070,32 +1068,22 @@ with gr.Blocks(theme=gr.themes.Soft()) as ui:
             yield "Starting...", None, " Starting app generation...", "Initializing"
             
             # Update progress
-            for i in progress.tqdm(range(5)):
-                if i == 0:
-                    msg = " Analyzing requirements..."
-                    stream_handler.update(msg, "Analyzing")
-                    yield None, None, stream_handler.output[-1], "Analyzing"
-                elif i == 1:
-                    msg = " Generating architecture..."
-                    stream_handler.update(msg, "Designing")
-                    yield None, None, stream_handler.output[-1], "Designing"
-                elif i == 2:
-                    msg = " Creating project structure..."
-                    stream_handler.update(msg, "Creating")
-                    yield None, None, stream_handler.output[-1], "Creating"
-                elif i == 3:
-                    msg = " Implementing features..."
-                    stream_handler.update(msg, "Implementing")
-                    yield None, None, stream_handler.output[-1], "Implementing"
-                elif i == 4:
-                    msg = " Finalizing..."
-                    stream_handler.update(msg, "Finalizing")
-                    yield None, None, stream_handler.output[-1], "Finalizing"
-                time.sleep(1)  # Simulate work
+            phases = [
+                (" Analyzing requirements...", "Analyzing"),
+                (" Generating architecture...", "Designing"),
+                (" Creating project structure...", "Creating"),
+                (" Implementing features...", "Implementing"),
+                (" Finalizing...", "Finalizing")
+            ]
+            
+            for msg, status in progress.tqdm(phases):
+                stream_handler.update(msg, status)
+                yield None, None, stream_handler.output[-1], status
+                await asyncio.sleep(1)  # Non-blocking sleep
             
             # Generate the app
             stream_handler.update(" Running AI Flow system...", "Processing")
-            result = asyncio.run(app_generator(requirements))
+            result = await app_generator(requirements)
             
             # Extract download path and logs
             download_path = None
@@ -1109,6 +1097,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as ui:
                     logs.append(line)
                     stream_handler.update(line.split(" - ")[-1], "Processing")
                     yield None, None, "\n".join(stream_handler.output), "Processing"
+                await asyncio.sleep(0.1)  # Small delay to prevent UI freezing
             
             if download_path and Path(download_path).exists():
                 stream_handler.update(" App generated successfully! Download is ready.", "Complete")
@@ -1123,7 +1112,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as ui:
             logger.error(error_msg)
             stream_handler.update(error_msg, "Error")
             yield error_msg, None, "\n".join(stream_handler.output), "Error"
-    
+
     def cancel_generation():
         """Cancel the current generation process."""
         stream_handler.update(" Generation cancelled by user", "Cancelled")