Update app.py

app.py

@@ -219,3 +219,31 @@ def run_process(file, question, model_choice, temperature, top_p, external_api_u
 
     except Exception as e:
         return f"ERROR: {type(e).__name__}: {e}", None
+
+    # ================== UI ==================
+def main():
+    with gr.Blocks(title="OCR Multi-Agent System") as demo:
+        file = gr.File(label="Upload PDF/Image")
+        question = gr.Textbox(label="Prompt", lines=2)
+        model_choice = gr.Dropdown(choices=[*INTERNAL_MODEL_MAP.keys(), EXTERNAL_MODEL_NAME],
+                                   value="Gemini 2.5 Flash", label="Model")
+        temperature = gr.Slider(0.0, 2.0, value=0.2, step=0.05)
+        top_p = gr.Slider(0.0, 1.0, value=0.95, step=0.01)
+        external_api_url = gr.Textbox(label="External API URL", visible=False)
+        output_text = gr.Code(label="Output", language="json")
+        run_btn = gr.Button("🚀 Process")
+
+        run_btn.click(
+            run_process,
+            inputs=[file, question, model_choice, temperature, top_p, external_api_url],
+            outputs=[output_text, gr.State()]
+        )
+
+    return demo
+
+demo = main()
+
+if __name__ == "__main__":
+    import os
+    os.system("which gs || echo '⚠️ ghostscript (gs) not found in PATH'")
+    demo.launch()