Update app.py

app.py

@@ -3,6 +3,9 @@ import os
 import json
 os.environ["OPENAI_API_KEY"] = os.getenv('api_key')
 
+from google import genai
+from google.genai import types
+
 import math
 import types
 import uuid
@@ -16,6 +19,16 @@ from langgraph_bigtool.utils import (
     convert_positional_only_function_to_tool
 )
 
+MODEL_ID = "gemini-2.0-flash-exp"
+from google import genai
+client = genai.Client(api_key=os.getenv('api_g_key'))
+
+def llm_response(text):
+  response = client.models.generate_content(
+    model=MODEL_ID,
+    contents= text)
+  return response.text
+
 # Collect functions from `math` built-in
 all_tools = []
 for function_name in dir(math):
@@ -113,7 +126,9 @@ def pvsnp(problem):
     print (output)'''
     output = agent.invoke({"messages": "Use tools to answer: "+problem})
     answer = output['messages'][5]
-    return answer
+    final_answer = llm_response(f'''Given the problem: {problem} and the agent response: {answer}, come up with a user friendly explanation that highlights the answer along
+    with the tools leveraged.''')
+    return final_answer
                                       
 
 iface = gr.Interface(