Added streaming output with real-time progress updates and status indicators

app.py

@@ -981,8 +981,24 @@ def app_generator(requirements: str):
     finally:
         loop.close()
 
+class StreamHandler:
+    """Handles streaming output for the Gradio interface."""
+    
+    def __init__(self):
+        self.output = []
+        self.current_status = ""
+    
+    def update(self, message: str, status: str = None):
+        """Update the output stream."""
+        self.output.append(message)
+        if status:
+            self.current_status = status
+        return "\n".join(self.output), self.current_status
+
 # Gradio UI
 with gr.Blocks(theme=gr.themes.Soft()) as ui:
+    stream_handler = StreamHandler()
+    
     gr.Markdown("# Autonomous App Generator with AI Flow")
     gr.Markdown("""
     ## Instructions
@@ -1009,7 +1025,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as ui:
                 placeholder="Describe the app you want to build...",
                 lines=10
             )
-            generate_button = gr.Button("Generate App", variant="primary")
+            with gr.Row():
+                generate_button = gr.Button("Generate App", variant="primary")
+                cancel_button = gr.Button("Cancel", variant="stop")
+            
+            status = gr.Textbox(
+                label="Status",
+                value="Ready",
+                interactive=False
+            )
             
         with gr.Column(scale=6):
             with gr.Tabs():
@@ -1023,21 +1047,48 @@ with gr.Blocks(theme=gr.themes.Soft()) as ui:
                         label="Download Generated App",
                         interactive=False
                     )
-                with gr.TabItem("Logs"):
+                with gr.TabItem("Live Log"):
                     log_output = gr.Markdown(
                         label="Generation Logs",
-                        value="Logs will appear here..."
+                        value="Logs will appear here...",
+                        autoscroll=True
                     )
     
-    def process_output(requirements):
+    def stream_output(requirements, progress=gr.Progress()):
+        """Stream the output during app generation."""
         try:
-            # Initialize log output
-            log_output.value = "Starting app generation...\n"
+            # Initialize
+            stream_handler.update("Starting app generation...", "Initializing")
+            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
             
             # Generate the app
             result = asyncio.run(app_generator(requirements))
             
-            # Extract download path from the result
+            # Extract download path and logs
             download_path = None
             logs = []
             
@@ -1046,27 +1097,38 @@ with gr.Blocks(theme=gr.themes.Soft()) as ui:
                     download_path = line.split(": ")[1].strip()
                 if line.startswith("2025-") or line.startswith("INFO") or line.startswith("ERROR"):
                     logs.append(line)
-            
-            # Update logs
-            log_output.value = "\n".join(logs)
+                    stream_handler.update(line)
+                    yield None, None, "\n".join(stream_handler.output), "Processing"
             
             if download_path and Path(download_path).exists():
-                return result, download_path, "\n".join(logs)
+                stream_handler.update(" App generated successfully!", "Complete")
+                yield result, download_path, "\n".join(stream_handler.output), "Complete"
             else:
-                error_msg = "Failed to generate download file"
-                log_output.value += f"\nERROR: {error_msg}"
-                return result, None, "\n".join(logs)
+                error_msg = " Failed to generate download file"
+                stream_handler.update(error_msg, "Error")
+                yield result, None, "\n".join(stream_handler.output), "Error"
                 
         except Exception as e:
-            error_msg = f"Failed to generate app: {str(e)}"
+            error_msg = f" Failed to generate app: {str(e)}"
             logger.error(error_msg)
-            log_output.value += f"\nERROR: {error_msg}"
-            return error_msg, None, log_output.value
+            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")
+        return "Generation cancelled", None, "\n".join(stream_handler.output), "Cancelled"
     
     generate_button.click(
-        process_output,
+        stream_output,
         inputs=[requirements_input],
-        outputs=[output, file_output, log_output]
+        outputs=[output, file_output, log_output, status],
+        show_progress=True
+    )
+    
+    cancel_button.click(
+        cancel_generation,
+        outputs=[output, file_output, log_output, status]
     )
 
 # Run the Gradio app