Update app.py

app.py

@@ -9,24 +9,32 @@ from azure.ai.projects import AIProjectClient
 from azure.identity import ClientSecretCredential
 from azure.ai.agents.models import ListSortOrder
 
-# Azure config
+# Azure config (from HF Space secrets) 
 PROJECT_ENDPOINT = os.environ["PROJECT_ENDPOINT"]
 AGENT_ID = os.environ["AGENT_ID"]
 AZURE_TENANT_ID = os.environ["AZURE_TENANT_ID"]
 AZURE_CLIENT_ID = os.environ["AZURE_CLIENT_ID"]
 AZURE_CLIENT_SECRET = os.environ["AZURE_CLIENT_SECRET"]
 
-# Azure client (singleton)
+# Azure client (singleton) 
 credential = ClientSecretCredential(
     tenant_id=AZURE_TENANT_ID,
     client_id=AZURE_CLIENT_ID,
     client_secret=AZURE_CLIENT_SECRET,
 )
 project = AIProjectClient(credential=credential, endpoint=PROJECT_ENDPOINT)
-agent = project.agents.get_agent(AGENT_ID)
 
+# Lazy agent fetch (so app boots even if Azure is slow)
+_agent = None
 
-# Chat logic 
+def get_agent():
+    global _agent
+    if _agent is None:
+        _agent = project.agents.get_agent(AGENT_ID)
+    return _agent
+
+
+# Chat logic
 def respond(user_message: str, history: list, thread_state: dict | None):
     """
     Gradio calls this for every user message.
@@ -44,6 +52,8 @@ def respond(user_message: str, history: list, thread_state: dict | None):
     thread_id = thread_state["thread_id"]
 
     try:
+        agent = get_agent()
+
         # Send user message
         project.agents.messages.create(
             thread_id=thread_id,
@@ -123,15 +133,13 @@ CSS = """
 footer { display: none !important; }
 """
 
-with gr.Blocks(css=CSS, title="Agent Ken — PM Copilot", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Agent Ken — PM Copilot") as demo:
 
     gr.Markdown(DESCRIPTION)
 
     chatbot = gr.Chatbot(
         label="Agent Ken",
         height=500,
-        type="messages",
-        avatar_images=(None, "https://em-content.zobj.net/source/twitter/408/robot_1f916.png"),
         show_copy_button=True,
     )
 
@@ -157,7 +165,7 @@ with gr.Blocks(css=CSS, title="Agent Ken — PM Copilot", theme=gr.themes.Soft()
             inputs=msg_input,
         )
 
-    # ── Event bindings ────────────────────────────────────────────
+    # Event bindings 
     send_btn.click(
         fn=respond,
         inputs=[msg_input, chatbot, thread_state],
@@ -177,4 +185,4 @@ with gr.Blocks(css=CSS, title="Agent Ken — PM Copilot", theme=gr.themes.Soft()
 
 # Launch 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860, css=CSS, theme=gr.themes.Soft())