Update app.py

app.py

@@ -3,7 +3,6 @@ import os
 from agent import PilatopiaAgentSystem
 from langchain_core.messages import HumanMessage, AIMessage
 
-# Initialize system
 agent_system = None
 try:
     api_key = os.getenv("OPENAI_API_KEY")
@@ -12,7 +11,6 @@ try:
 except Exception as e:
     print(f"❌ System Error: {e}")
 
-# Global conversation state
 conversation_state = {
     "messages": [],
     "current_plan": "",
@@ -26,17 +24,14 @@ conversation_state = {
 }
 
 def chat_with_agent(message, history):
-    """Simple chat function for Gradio"""
     if not agent_system:
-        return "⚠️ System not available. Please check your API key."
+        return "System not available. Please check your API key."
     
     try:
-        # Add user message
         global conversation_state
         new_user_message = HumanMessage(content=message)
         conversation_state["messages"] = conversation_state["messages"] + [new_user_message]
         
-        # Reset for new processing
         conversation_state.update({
             "current_plan": "",
             "response_draft": "",
@@ -47,13 +42,11 @@ def chat_with_agent(message, history):
             "iteration_count": 0
         })
         
-        # Run agent system
         final_state = conversation_state
         for event in agent_system.graph.stream(conversation_state):
             for node_name, value in event.items():
                 final_state = value
         
-        # Update global state and return response
         conversation_state = final_state
         return final_state.get("final_response", "عذراً، حدث خطأ.")
         
@@ -75,7 +68,6 @@ def reset_chat():
         "plan_approved": False
     }
 
-# Create Gradio interface
 demo = gr.ChatInterface(
     fn=chat_with_agent,
     title="🏋️‍♀️ Pilatopia Customer Support",