Update app.py

app.py

@@ -2,87 +2,135 @@ import gradio as gr
 import ai_engine
 import zoho_client
 import time
+import json
 import sys
 
 # Force unbuffered output for logs
 sys.stdout.reconfigure(line_buffering=True)
 
-def process_pipeline(file):
-    print("--- NEW REQUEST STARTED ---")
-    logs = ""
+# ==========================================
+# 1. STEP 1: ANALYZE (OCR + AI)
+# ==========================================
+def step_analyze(file):
+    """
+    Runs OCR and AI extraction, but DOES NOT send to Zoho.
+    Returns the data for the user to edit.
+    """
+    logs = "--- PHASE 1: ANALYSIS ---\n"
     
+    if file is None:
+        yield None, None, "❌ Please upload a file first."
+        return
+
     # 1. OCR
     try:
-        print("Step 1: Starting OCR...")
         logs += "👁️ Scanning Document (OCR)...\n"
-        yield None, logs
-        
+        yield None, None, logs
         text, img, meta = ai_engine.perform_ocr(file)
         
         if not text:
-            print("Error: OCR returned empty text.")
-            logs += "❌ OCR Failed. No text found in document.\n"
-            yield None, logs
+            logs += "❌ OCR Failed. No text found.\n"
+            yield None, None, logs
             return
-        print(f"OCR Success. Text length: {len(text)}")
-        
     except Exception as e:
-        print(f"OCR Exception: {e}")
-        logs += f"❌ System Error during OCR: {e}\n"
-        yield None, logs
+        logs += f"❌ OCR Error: {e}\n"
+        yield None, None, logs
         return
 
     # 2. AI Extraction
     try:
-        print("Step 2: Running AI Extraction...")
-        logs += "🧠 AI Analyzing & Classifying (this takes ~10s)...\n"
-        yield img, logs
+        logs += "🧠 AI Analyzing & Classifying (1.5B Model)...\n"
+        yield img, None, logs
         
         ai_output = ai_engine.extract_intelligent_json(text, meta)
-        print(f"AI Output: {ai_output}")
         
         doc_type = ai_output.get('doc_type', 'unknown')
-        logs += f"🤖 Detected: {doc_type.upper()}\n"
-        logs += "----------------------------------\n"
-        yield img, logs
+        logs += f"🤖 Detected Type: {doc_type.upper()}\n"
+        logs += "✅ Analysis Complete. Please REVIEW the JSON below before syncing.\n"
+        
+        # Return Image, JSON Data for Editor, and Logs
+        # We dump to string for the Code editor, nicely formatted
+        json_str = json.dumps(ai_output, indent=4)
+        yield img, json_str, logs
         
     except Exception as e:
-        print(f"AI Exception: {e}")
-        logs += f"❌ AI Analysis Failed: {e}\n"
-        yield img, logs
-        return
+        logs += f"❌ AI Error: {e}\n"
+        yield img, None, logs
+
+# ==========================================
+# 2. STEP 2: SYNC (Zoho Push)
+# ==========================================
+def step_sync(json_data_str):
+    """
+    Takes the EDITED JSON from the UI and pushes it to Zoho.
+    """
+    logs = "\n--- PHASE 2: ZOHO SYNC ---\n"
+    yield logs
     
-    # 3. Orchestration
-    print("Step 3: Starting Zoho Orchestration...")
+    # 1. Parse Edited JSON
+    try:
+        if not json_data_str:
+            yield logs + "❌ Error: No data to sync. Please analyze a document first."
+            return
+            
+        ai_output = json.loads(json_data_str)
+    except Exception as e:
+        yield logs + f"❌ Invalid JSON Format: {e}\n(Did you delete a comma or bracket?)"
+        return
+
+    # 2. Execute Zoho Logic
     try:
         # Consume the generator from zoho_client
         for update in zoho_client.route_and_execute(ai_output):
-            print(f"Zoho Update: {update.strip()}")
             logs += update
-            yield img, logs
-            time.sleep(0.1) # Prevent UI flooding
+            yield logs
+            time.sleep(0.1) # Visual flow
             
     except Exception as e:
-        print(f"Orchestrator Exception: {e}")
         logs += f"\n❌ Orchestrator Error: {e}"
-        yield img, logs
+        yield logs
 
-# --- UI SETUP ---
-with gr.Blocks(title="Zoho Agentic") as demo:
-    gr.Markdown("## ⚡ Zoho Agentic Orchestrator (Debug Mode)")
-    gr.Markdown("If the UI hangs, check the 'Logs' tab in the Hugging Face Space.")
+# ==========================================
+# 3. UI LAYOUT (Human-in-the-Loop)
+# ==========================================
+with gr.Blocks(title="Zoho Agentic HITL", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 🕵️‍♂️ Zoho Agent: Human-in-the-Loop")
+    gr.Markdown("1. **Upload** & Analyze. 2. **Review/Edit** the extracted data. 3. **Confirm** to Sync.")
     
     with gr.Row():
-        f_in = gr.File(label="Upload Document")
-        btn = gr.Button("Start Processing", variant="primary")
-        
-    with gr.Row():
-        out_img = gr.Image(label="Document View", height=400)
-        out_log = gr.Code(label="Live Execution Logs", language="shell")
+        # LEFT COLUMN: Input & Visuals
+        with gr.Column(scale=1):
+            f_in = gr.File(label="1. Upload Document")
+            btn_analyze = gr.Button("🔍 Analyze Document", variant="secondary")
+            out_img = gr.Image(label="Document Preview", height=500)
+
+        # RIGHT COLUMN: Data Editor & logs
+        with gr.Column(scale=1):
+            # The Editor
+            gr.Markdown("### 2. Review Extracted Data")
+            json_editor = gr.Code(label="JSON Editor (Editable)", language="json", interactive=True, lines=20)
+            
+            # The Action
+            btn_sync = gr.Button("🚀 Confirm & Sync to Zoho", variant="primary")
+            
+            # The Logs
+            out_log = gr.Code(label="Execution Logs", language="shell", lines=10)
+
+    # --- EVENT WIRING ---
+    
+    # Click Analyze -> Updates Image, JSON Editor, Logs
+    btn_analyze.click(
+        fn=step_analyze, 
+        inputs=[f_in], 
+        outputs=[out_img, json_editor, out_log]
+    )
     
-    btn.click(process_pipeline, f_in, [out_img, out_log])
+    # Click Sync -> Reads JSON Editor -> Updates Logs
+    btn_sync.click(
+        fn=step_sync,
+        inputs=[json_editor],
+        outputs=[out_log]
+    )
 
 if __name__ == "__main__":
-    # CRITICAL FIX: ssr_mode=False prevents blank screens on some browsers
-    print("Launching Gradio Server...")
     demo.launch(ssr_mode=False)