Update app.py

app.py

@@ -95,7 +95,6 @@ class Me:
         self.session_log = []
         Path("chat_logs").mkdir(exist_ok=True)
 
-        # Download resume and summary
         gdown.download("https://drive.google.com/uc?id=1xz2RowkImpI8odYv8zvKdlRHaKfILn40", "linkedin.pdf", quiet=False)
         reader = PdfReader("linkedin.pdf")
         self.linkedin = "".join(page.extract_text() or "" for page in reader.pages)
@@ -148,17 +147,36 @@ If you can't answer something, call `record_unknown_question`. If a user seems i
             model="gpt-4o",
             messages=messages,
             tools=tools,
-            stream=True
+            stream=False
         )
 
-        full_response = ""
-        for chunk in response:
-            delta = chunk.choices[0].delta
-            if hasattr(delta, "content") and delta.content:
-                full_response += delta.content
-                yield full_response
+        reply = response.choices[0].message
+
+        if reply.tool_calls:
+            messages.append(reply)
+            tool_results = self.handle_tool_call(reply.tool_calls)
+            messages.extend(tool_results)
+
+            follow_up = "✅ I've saved that info. Let me know if you'd like to ask more questions."
+            self.session_log.append({"role": "assistant", "content": follow_up})
+            yield follow_up
+        else:
+            stream = self.openai.chat.completions.create(
+                model="gpt-4o",
+                messages=messages,
+                tools=tools,
+                stream=True
+            )
+
+            full_response = ""
+            for chunk in stream:
+                delta = chunk.choices[0].delta
+                if hasattr(delta, "content") and delta.content:
+                    full_response += delta.content
+                    yield full_response
+
+            self.session_log.append({"role": "assistant", "content": full_response})
 
-        self.session_log.append({"role": "assistant", "content": full_response})
         self.save_session_log()
 
     def save_session_log(self):