Update app.py

app.py

@@ -1,36 +1,51 @@
 import gradio as gr
 import ai_engine
 import zoho_client
+import time
 
 def process_pipeline(file):
-    # 1. OCR & Metadata
+    logs = ""
+    
+    # 1. OCR (Yield immediate feedback)
+    logs += "👁️ Scanning Document (OCR)...\n"
+    yield None, logs
     text, img, meta = ai_engine.perform_ocr(file)
-    if not text: return None, "❌ OCR Failed / No Text Found"
     
-    # 2. AI Classification (With Metadata)
+    if not text:
+        logs += "❌ OCR Failed. No text found.\n"
+        yield None, logs
+        return
+
+    # 2. AI Extraction
+    logs += "🧠 AI Analyzing & Classifying...\n"
+    yield img, logs
     ai_output = ai_engine.extract_intelligent_json(text, meta)
     
-    # 3. Execution
-    log = f"📂 Filename: {meta.get('filename')}\n"
-    log += f"🤖 AI Classification: {ai_output.get('doc_type')}\n"
-    log += "----------------------------------\n"
-    
-    result = zoho_client.route_and_execute(ai_output)
+    logs += f"🤖 Detected: {ai_output.get('doc_type')}\n"
+    logs += "----------------------------------\n"
+    yield img, logs
     
-    return img, log + result
+    # 3. Orchestration (Consuming the Stream)
+    # We loop through the generator from zoho_client
+    for update in zoho_client.route_and_execute(ai_output):
+        logs += update # Append new log line
+        yield img, logs # Update UI
+        time.sleep(0.2) # Tiny visual delay for readability
 
-with gr.Blocks(title="Zoho Intelligent Agent") as demo:
-    gr.Markdown("## 🧠 Zoho Autonomous Agent")
-    gr.Markdown("The system uses **Filename + OCR Content** to decide the route.")
+# UI Setup
+with gr.Blocks(title="Zoho Agentic") as demo:
+    gr.Markdown("## ⚡ Zoho Agentic Orchestrator (Live Stream)")
+    gr.Markdown("Upload a file. Watch the Agent resolve dependencies in real-time.")
     
     with gr.Row():
-        f_in = gr.File(label="Upload Document (Invoice, Bill, PO, Receipt)")
-        btn = gr.Button("Analyze & Sync", variant="primary")
+        f_in = gr.File(label="Upload Document")
+        btn = gr.Button("Start Processing", variant="primary")
         
-    out_img = gr.Image(label="Preview", height=300)
-    out_res = gr.Textbox(label="Orchestrator Log", lines=8)
+    out_img = gr.Image(label="Document View", height=400)
+    # Using a Code block for logs looks cooler/cleaner
+    out_log = gr.Code(label="Agent Execution Logs", language="shell")
     
-    btn.click(process_pipeline, f_in, [out_img, out_res])
+    btn.click(process_pipeline, f_in, [out_img, out_log])
 
 if __name__ == "__main__":
     demo.launch()