Fix: use uvicorn entry point & gr.Code for json output

app.py

@@ -679,7 +679,7 @@ def health():
 
 def ui_chat_wrapper(message, model_id, pack_id, max_tokens, temperature):
     if not message.strip():
-        return "Please enter a message.", {}
+        return "Please enter a message.", "{}"
     
     msgs = [ChatMessage(role="user", content=message)]
     req = ChatRequest(
@@ -697,9 +697,10 @@ def ui_chat_wrapper(message, model_id, pack_id, max_tokens, temperature):
     try:
         res = chat_completions(req, request=FakeRequest(), _rate={"ip": "gradio_ui"})
         answer = res["choices"][0]["message"]["content"]
-        return answer, res
+        return answer, json.dumps(res, indent=2)
     except Exception as e:
-        return f"Error: {str(e)}", {"error": str(e), "trace": traceback.format_exc()}
+        err_res = {"error": str(e), "trace": traceback.format_exc()}
+        return f"Error: {str(e)}", json.dumps(err_res, indent=2)
 
 def ui_compile_text(text, bandwidth):
     if not text.strip():
@@ -781,7 +782,6 @@ with gr.Blocks(title="Kalpanā AI — Testing Suite & Studio", theme=gr.themes.S
     )
     
     with gr.Tabs():
-        # TAB 1: CHAT & TEST
         with gr.TabItem("💬 Chat & Inference Test"):
             gr.Markdown("### Test RIF-Enhanced Chat Inference with any LLM")
             with gr.Row():
@@ -802,7 +802,7 @@ with gr.Blocks(title="Kalpanā AI — Testing Suite & Studio", theme=gr.themes.S
                     
                 with gr.Column(scale=3):
                     bot_output = gr.Textbox(lines=8, label="LLM Assistant Response", interactive=False)
-                    json_output = gr.JSON(label="Full API Response & RIF Context Metrics")
+                    json_output = gr.Code(language="json", label="Full API Response & RIF Context Metrics")
 
             submit_btn.click(
                 fn=ui_chat_wrapper,
@@ -810,7 +810,6 @@ with gr.Blocks(title="Kalpanā AI — Testing Suite & Studio", theme=gr.themes.S
                 outputs=[bot_output, json_output]
             )
 
-        # TAB 2: KNOWLEDGE PACK STUDIO
         with gr.TabItem("📚 Knowledge Pack Studio"):
             gr.Markdown("### Compile Documents into Portable .kp Knowledge Packs")
             with gr.Row():
@@ -847,7 +846,6 @@ with gr.Blocks(title="Kalpanā AI — Testing Suite & Studio", theme=gr.themes.S
             packs_md = gr.Markdown("Click Refresh to load active packs.")
             refresh_btn.click(fn=ui_refresh_packs, inputs=[], outputs=[packs_md])
 
-        # TAB 3: REGISTER CUSTOM LLM PROVIDER
         with gr.TabItem("🔌 Register Custom LLM Provider"):
             gr.Markdown("### Bring Your Own API Key (Groq, OpenAI, Together, Cerebras, OpenRouter)")
             with gr.Row():
@@ -870,7 +868,6 @@ with gr.Blocks(title="Kalpanā AI — Testing Suite & Studio", theme=gr.themes.S
                 outputs=[reg_status]
             )
 
-        # TAB 4: CONTEXT EXTENSION BENCHMARK MATRIX
         with gr.TabItem("📊 Verified Context Extension Matrix"):
             gr.Markdown(
                 """
@@ -894,11 +891,14 @@ with gr.Blocks(title="Kalpanā AI — Testing Suite & Studio", theme=gr.themes.S
                 """
             )
 
-app = gr.mount_gradio_app(app, demo, path="/ui")
-
 @app.get("/", include_in_schema=False)
 def root_redirect():
     return RedirectResponse(url="/docs")
 
+# Mount Gradio UI on FastAPI app
+app = gr.mount_gradio_app(app, demo, path="/ui")
+
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+