Update app.py

app.py

@@ -18,6 +18,7 @@ def fetch_catalog(query):
         return "Search service offline."
 
 def librarian_agent(message, history, session_state):
+    # Standardize state initialization
     if session_state is None:
         session_state = {"verified": False, "age": None}
 
@@ -27,7 +28,7 @@ def librarian_agent(message, history, session_state):
         if age_input:
             age = int(age_input)
             if age < 13:
-                reply = "🛡️ Safety: You must be 13+ to use this service."
+                reply = "🛡️ Safety: Access restricted for users under 13."
             else:
                 session_state["age"] = age
                 session_state["verified"] = True
@@ -35,50 +36,51 @@ def librarian_agent(message, history, session_state):
         else:
             reply = "👋 Please enter your age as a number to start."
         
-        # GRADIO 6 FORMAT: Role/Content Dictionaries
+        # history is a list of {"role": "...", "content": "..."}
         history.append({"role": "user", "content": message})
         history.append({"role": "assistant", "content": reply})
         return history, session_state
 
     # --- PHASE 2: RAG PIPELINE ---
     catalog_context = fetch_catalog(message)
-    messages = [
+    
+    # Building the internal LLM prompt
+    llm_messages = [
         {"role": "system", "content": f"You are a librarian for a {session_state['age']} year old. Use: {catalog_context}"},
         {"role": "user", "content": message}
     ]
 
     try:
-        output = client.chat_completion(messages=messages, max_tokens=1000)
+        output = client.chat_completion(messages=llm_messages, max_tokens=1000)
         bot_res = output.choices[0].message.content
     except Exception as e:
-        bot_res = f"Error: {str(e)}"
+        bot_res = f"Service Error: {str(e)}"
 
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": bot_res})
     return history, session_state
 
-# --- GRADIO 6 UI ---
+# --- UI BLOCK (Minimalist) ---
 with gr.Blocks() as demo:
-    gr.Markdown("## 📚 AI Librarian (Gradio 6 Build)")
+    gr.Markdown("## 📚 AI Librarian Agent")
     
     state = gr.State(None)
     
-    # NEW: Initial value must follow {"role": "...", "content": "..."} format
-    initial_msg = [
-        {"role": "assistant", "content": "👋 Welcome! Please enter your **age** to begin."}
-    ]
-    chatbot = gr.Chatbot(label="Librarian Consultation", value=initial_msg, type="messages")
+    # We initialize with a system message in the new dict format
+    # We REMOVE 'type="messages"' to avoid the keyword error
+    chatbot = gr.Chatbot(
+        label="Librarian Consultation",
+        value=[{"role": "assistant", "content": "👋 Welcome! Please enter your **age** to begin."}]
+    )
     
     msg = gr.Textbox(label="Input", placeholder="Enter age (e.g. 25)")
-    clear = gr.Button("Clear")
 
     def respond(user_input, chat_history, current_state):
+        # Gradio automatically detects the format if 'type' is omitted
         updated_history, updated_state = librarian_agent(user_input, chat_history, current_state)
         return "", updated_history, updated_state
 
     msg.submit(respond, [msg, chatbot, state], [msg, chatbot, state])
-    clear.click(lambda: (None, initial_msg, None), None, [msg, chatbot, state])
 
-# NEW: Theme moved to launch()
-if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft())
+# Launch without passing theme to Blocks, handle it here if needed
+demo.launch()