Update app.py

app.py

@@ -100,18 +100,27 @@ for tool in sympy_tools:
 builder = create_agent(llm, tool_registry)
 agent = builder.compile(store=store)
 
-
-
-                                         
+def pvsnp(problem):
+    output = []
+    for step in agent.stream(
+    {"messages": "Use tools to answer:"+problem},
+    stream_mode="updates",
+):
+    for _, update in step.items():
+        for message in update.get("messages", []):
+            message.pretty_print()
+            output.append(message.pretty_print())
+    return output
+                                      
 
 iface = gr.Interface(
     fn=pvsnp, 
     inputs=gr.Textbox(label="What problem would you like to classify as P or NP?"), 
     outputs=gr.Markdown(label="Agent's response"),  # Output as HTML
-    title="PolyProb",
-    description="PolyProb is an advanced AI agent that guides users through the intricate maze of computational complexity. This agent scrutinizes problem descriptions with sophisticated LLM prompts and symbolic reasoning. It classifies problems into categories such as P, NP, NP-complete, NP-hard, or beyond (e.g., PSPACE, EXPTIME), while providing clear, concise explanations of its reasoning. As part of AI Quotient’s Millennium Math Challenge, it is the first step towards solving the P vs NP problem.",
+    title="PolyMath",
+    description="PolyMath is an advanced AI agent that guides users through the intricate maze of computational complexity. This agent scrutinizes problem descriptions with sophisticated LLM prompts and symbolic reasoning. It classifies problems into categories such as P, NP, NP-complete, NP-hard, or beyond (e.g., PSPACE, EXPTIME), while providing clear, concise explanations of its reasoning. As part of AI Quotient’s Millennium Math Challenge, it is the first step towards solving the P vs NP problem.",
     theme = gr.themes.Ocean(),
-    examples = ["Can you find a path that visits each node exactly once in a given graph?", "How efficiently can two nXn matrices be multiplied?", "Is there a subset of given numbers that sums to a target value?"]
+    examples = ["Simplify x*2+2x+1"]
 )
 
 # Launch the app