Update app.py

app.py

@@ -12,14 +12,15 @@ import zipfile
 
 load_dotenv(override=True)
 
-# ─── Pushover Notifications ───────────────────────────────────────────────
+# ─── Ensure chat_logs Folder Exists ───
+logs_path = Path("chat_logs")
+logs_path.mkdir(exist_ok=True)
 
+# ─── Pushover Notifications ───────────
 def push(text):
     try:
-        Path("chat_logs").mkdir(exist_ok=True)
-        keep_path = Path("chat_logs/.keep")
-        if not keep_path.exists():
-            keep_path.touch()
+        keep_path = logs_path / ".keep"
+        keep_path.touch(exist_ok=True)
         requests.post(
             "https://api.pushover.net/1/messages.json",
             data={
@@ -31,15 +32,10 @@ def push(text):
     except Exception as e:
         print(f"Pushover error: {e}")
 
-# ─── Tool Functions ────────────────────────────────────────────────────────
-
+# ─── Tool Functions ───────────────────
 def record_user_details(email, name="Name not provided", notes="not provided"):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    filename = f"chat_logs/session_{timestamp}.json"
-    latest_log = "\n".join([
-        f"{entry['role'].capitalize()}: {entry['content'][:200]}"
-        for entry in me.session_log[-6:]
-    ])
+    filename = logs_path / f"session_{timestamp}.json"
     with open(filename, "w", encoding="utf-8") as f:
         json.dump(me.session_log, f, indent=2)
     msg = f"[New Contact]\nName: {name}\nEmail: {email}\nNotes: {notes}\n\n🔗 View log: {filename}"
@@ -48,58 +44,48 @@ def record_user_details(email, name="Name not provided", notes="not provided"):
 
 def record_unknown_question(question):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    filename = f"chat_logs/session_{timestamp}.json"
-    latest_log = "\n".join([
-        f"{entry['role'].capitalize()}: {entry['content'][:200]}"
-        for entry in me.session_log[-6:]
-    ])
+    filename = logs_path / f"session_{timestamp}.json"
     with open(filename, "w", encoding="utf-8") as f:
         json.dump(me.session_log, f, indent=2)
     msg = f"[Unknown Question]\nQ: {question}\n\n🔗 View log: {filename}"
     push(msg)
     return {"recorded": "ok"}
 
-record_user_details_json = {
-    "name": "record_user_details",
-    "description": "Use this tool to record that a user is interested in being in touch and provided an email address",
-    "parameters": {
-        "type": "object",
-        "properties": {
-            "email": {"type": "string"},
-            "name": {"type": "string"},
-            "notes": {"type": "string"}
-        },
-        "required": ["email"],
-        "additionalProperties": False
-    }
-}
-
-record_unknown_question_json = {
-    "name": "record_unknown_question",
-    "description": "Record a question that couldn't be answered",
-    "parameters": {
-        "type": "object",
-        "properties": {
-            "question": {"type": "string"}
-        },
-        "required": ["question"],
-        "additionalProperties": False
-    }
-}
-
 tools = [
-    {"type": "function", "function": record_user_details_json},
-    {"type": "function", "function": record_unknown_question_json}
+    {"type": "function", "function": {
+        "name": "record_user_details",
+        "description": "Record user contact information.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "email": {"type": "string"},
+                "name": {"type": "string"},
+                "notes": {"type": "string"}
+            },
+            "required": ["email"]
+        }
+    }},
+    {"type": "function", "function": {
+        "name": "record_unknown_question",
+        "description": "Record questions unable to be answered.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "question": {"type": "string"}
+            },
+            "required": ["question"]
+        }
+    }}
 ]
 
-# ─── Core Chatbot Class ────────────────────────────────────────────────────
-
+# ─── Core Chatbot Class ───────────────
 class Me:
     def __init__(self):
         self.openai = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
         self.name = "Jacob Isaacson"
         self.session_log = []
-        Path("chat_logs").mkdir(exist_ok=True)
+
+        logs_path.mkdir(exist_ok=True)
 
         gdown.download("https://drive.google.com/uc?id=1xz2RowkImpI8odYv8zvKdlRHaKfILn40", "linkedin.pdf", quiet=False)
         reader = PdfReader("linkedin.pdf")
@@ -112,14 +98,11 @@ class Me:
         self.archive_logs()
 
     def system_prompt(self):
-        return f"""You are acting as {self.name}. You're answering questions on {self.name}'s website about his career, experience, and skills.
-Be professional and conversational, as if talking to a potential employer or client.
-
+        return f"""You are acting as {self.name}. You're answering questions about {self.name}'s career, experience, and skills.
+Be professional and conversational.
 If you can't answer something, call `record_unknown_question`. If a user seems interested, ask for their email and use `record_user_details`.
-
 ## Summary:
 {self.summary}
-
 ## LinkedIn Profile:
 {self.linkedin}
 """
@@ -127,105 +110,64 @@ If you can't answer something, call `record_unknown_question`. If a user seems i
     def handle_tool_call(self, tool_calls):
         results = []
         for tool_call in tool_calls:
-            tool_name = tool_call.function.name
+            func = globals()[tool_call.function.name]
             arguments = json.loads(tool_call.function.arguments)
-            tool = globals().get(tool_name)
-            result = tool(**arguments) if tool else {}
-            results.append({"role": "tool", "tool_call_id": tool_call.id, "content": json.dumps(result)})
+            result = func(**arguments)
+            results.append({
+                "role": "tool",
+                "tool_call_id": tool_call.id,
+                "content": json.dumps(result)
+            })
         return results
 
     def chat_stream(self, message, history):
         messages = [{"role": "system", "content": self.system_prompt()}]
 
         for msg in history:
-            if isinstance(msg, dict) and msg.get("role") in ["user", "assistant"]:
-                messages.append(msg)
+            messages.append(msg)
 
         messages.append({"role": "user", "content": message})
         self.session_log.append({"role": "user", "content": message})
 
         response = self.openai.chat.completions.create(
-            model="gpt-4o",
-            messages=messages,
-            tools=tools,
-            stream=False
+            model="gpt-4o", messages=messages, tools=tools, stream=False
         )
 
         reply = response.choices[0].message
+        full_response = reply.content or ""
+
+        self.session_log.append({"role": "assistant", "content": full_response})
+        self.save_session_log()
 
-        if reply.tool_calls:
-            tool_results = self.handle_tool_call(reply.tool_calls)
-            messages.append(reply)
-            messages.extend(tool_results)
-
-            final_response = self.openai.chat.completions.create(
-                model="gpt-4o",
-                messages=messages,
-                tools=tools,
-                stream=True
-            )
-
-            full_response = ""
-            for chunk in final_response:
-                delta = chunk.choices[0].delta
-                if hasattr(delta, "content") and delta.content:
-                    full_response += delta.content
-                    yield full_response
-
-            full_response += "\n\n💬 Let me know if you’d like to follow up or need help connecting with Jacob."
-            self.session_log.append({"role": "assistant", "content": full_response})
-            self.save_session_log()
-        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
-
-            full_response += "\n\n💬 Let me know if you’d like to follow up or need help connecting with Jacob."
-            self.session_log.append({"role": "assistant", "content": full_response})
-            self.save_session_log()
+        yield full_response + "\n\n💬 Let me know if you’d like to follow up or need help connecting with Jacob."
 
     def save_session_log(self):
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"chat_logs/session_{timestamp}.json"
+        filename = logs_path / f"session_{timestamp}.json"
         with open(filename, "w", encoding="utf-8") as f:
             json.dump(self.session_log, f, indent=2)
 
     def archive_logs(self):
-        zip_path = "chat_logs/weekly_archive.zip"
-        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as archive:
-            for log_file in Path("chat_logs").glob("session_*.json"):
+        zip_path = logs_path / "weekly_archive.zip"
+        with zipfile.ZipFile(zip_path, "w") as archive:
+            for log_file in logs_path.glob("session_*.json"):
                 archive.write(log_file, arcname=log_file.name)
 
-# ─── UI Interface ───────────────────────────────────────────────────────────
-
 me = Me()
 
 with gr.Blocks(title="Jacob Isaacson Chatbot") as iface:
     with gr.Row():
-        gr.Image("jacob.png", width=100, show_label=False)
-        gr.Markdown("### Chat with Jacob Isaacson\nAsk about Jacob's background, skills, or career. \n🛡️ *All chats are logged for improvement purposes.*")
-
+        gr.Markdown("# Chat with Jacob Isaacson")
     gr.ChatInterface(
         fn=me.chat_stream,
         chatbot=gr.Chatbot(show_copy_button=True),
         type="messages",
-        additional_inputs=[],
-        retry_btn=None,
-        undo_btn=None,
-        clear_btn=None,
-        chatbot_initial_message={"role": "assistant", "content": "Hello, my name is Jacob Isaacson. Please ask me any questions about my professional career and I will do my best to respond."}
+        chatbot_initial_message={
+            "role": "assistant",
+            "content": "Hello, my name is Jacob Isaacson. Ask any questions about my professional career!"
+        }
     )
 
-if __name__ == "__main__":
-    iface.launch()
+iface.launch()
+