Update app.py

app.py

@@ -18,8 +18,7 @@ def push(text):
     try:
         if not PUSHOVER_TOKEN or not PUSHOVER_USER:
             return
-        requests.post(
-            "https://api.pushover.net/1/messages.json",
+        requests.post("https://api.pushover.net/1/messages.json",
             data={"token": PUSHOVER_TOKEN, "user": PUSHOVER_USER, "message": text},
             timeout=5
         )
@@ -48,7 +47,7 @@ tools = [
                 "properties": {
                     "email": {"type": "string"},
                     "name": {"type": "string"},
-                    "notes": {"type": "string"}
+                    "notes": {"type": "string"},
                 },
                 "required": ["email"]
             }
@@ -89,9 +88,9 @@ class Me:
 
     def system_prompt(self):
         return f"""
-You are acting as {self.name}. You answer questions about his background, skills and experience.
-If unsure, use record_unknown_question.
-If user is interested, ask for an email and call record_user_details.
+You are acting as {self.name}. Answer questions about his background, skills and experience.
+If unsure, call record_unknown_question.
+If user is interested, ask for email and call record_user_details.
 
 Summary:
 {self.summary}
@@ -103,8 +102,8 @@ LinkedIn:
     def chat(self, message, history):
         messages = [{"role": "system", "content": self.system_prompt()}]
 
-        for role, text in history:
-            messages.append({"role": role, "content": text})
+        for msg in history:
+            messages.append(msg)
 
         messages.append({"role": "user", "content": message})
 
@@ -114,7 +113,7 @@ LinkedIn:
                 messages=messages,
                 tools=tools,
                 tool_choice="auto",
-                max_tokens=400
+                max_tokens=500
             )
 
             choice = response.choices[0]
@@ -122,9 +121,9 @@ LinkedIn:
             finish = choice.finish_reason
 
             if finish == "tool_calls":
-                for call in msg.tool_calls:
-                    fname = call.function.name
-                    args = json.loads(call.function.arguments)
+                for tool_call in msg.tool_calls:
+                    fname = tool_call.function.name
+                    args = json.loads(tool_call.function.arguments)
                     result = globals()[fname](**args)
                     messages.append({"role": "tool", "content": json.dumps(result)})
                 continue
@@ -136,37 +135,38 @@ me = Me()
 
 def respond(user_message, history):
     bot_reply = me.chat(user_message, history)
-    history.append(("user", user_message))
-    history.append(("assistant", bot_reply))
+    history.append({"role": "user", "content": user_message})
+    history.append({"role": "assistant", "content": bot_reply})
     return "", history
 
 
 with gr.Blocks(css="""
     body, .gradio-container { background-color: #0d0d0d; color: white; }
-    .gr-button { background-color: #ff4da6 !important; color: black !important; font-weight: 600 !important; }
+    .gr-button { background-color: #ff4da6 !important;
+                 color: black !important; font-weight: 600 !important; }
     .gr-button:hover { background-color: #ff1a8c !important; }
 """) as ui:
 
-    gr.Markdown("### Quick Options")
+    chatbot = gr.Chatbot(type="messages", height=420)
 
     with gr.Row():
-        btn_contact = gr.Button("Contact Info")
         btn_about = gr.Button("Who is Ayush?")
+        btn_contact = gr.Button("Contact Info")
         btn_projects = gr.Button("Latest Projects")
 
-    chatbot = gr.Chatbot(height=450)
     user_input = gr.Textbox(placeholder="Type your message...")
 
     user_input.submit(respond, [user_input, chatbot], [user_input, chatbot])
 
-    btn_contact.click(lambda: "What is Ayush Tyagi's contact information?", None, user_input)
-    btn_contact.click(respond, [user_input, chatbot], [user_input, chatbot])
-
     btn_about.click(lambda: "Who is Ayush Tyagi?", None, user_input)
     btn_about.click(respond, [user_input, chatbot], [user_input, chatbot])
 
+    btn_contact.click(lambda: "What is Ayush Tyagi's contact information?", None, user_input)
+    btn_contact.click(respond, [user_input, chatbot], [user_input, chatbot])
+
     btn_projects.click(lambda: "Show Ayush Tyagi’s latest projects.", None, user_input)
     btn_projects.click(respond, [user_input, chatbot], [user_input, chatbot])
 
+
 if __name__ == "__main__":
     ui.launch()