chore: update app.py

Signed-off-by: Suvaditya Mukherjee <[email protected]>

README.md

@@ -10,4 +10,7 @@ pinned: false
 license: apache-2.0
 ---
 
-A chatbot that helps you ask questions about Suvaditya!
+A chatbot that helps you ask questions about Suvaditya!
+
+Note: Sometimes, as always, the LLM may give wrong answers. Here's a link to my resume, if you'd like to go through it yourself!
+Get in touch with me through [X](https://x.com/halcyonrayes), [Gmail](mailto:[email protected]), [LinkedIn](https://www.linkedin.com/in/suvadityamukherjee), or [schedule a meeting with me here](https://cal.com/suvadityamuk)

app.py

@@ -99,6 +99,7 @@ def update_chat_history(chat_history, tool_query, query_results):
 
 if __name__ == "__main__":
     RESUME_DATA = """
+
     Suvaditya Mukherjee Email: [email protected]
     Portfolio: suvadityamuk.com Mobile: (213) 827-9733
     Github: github.com/suvadityamuk
@@ -278,6 +279,15 @@ if __name__ == "__main__":
 
     @spaces.GPU
     def rag_process(message, chat_history):
+        if not chat_history:
+            system_message = {
+                "role": "system",
+                "content": """You are an AI assistant focused on answering questions about Suvaditya's resume. 
+                Only provide information that is explicitly mentioned in the resume data. 
+                If you're unsure about any information, refuse to answer and direct users to suvadityamuk.com. 
+                Be accurate and concise in your responses. """
+            }
+            chat_history = [system_message]
         # wandb.init(
         #     project="resume-rag", 
         #     name="zerogpu-run",
@@ -355,8 +365,9 @@ if __name__ == "__main__":
         title="Suvaditya's Personal RAG, a space on ZeroGPU!",
         examples=["Where did Suvaditya complete his Bachelor's Degree?", "Where is Suvaditya currently working?"],
         description="Ask any question about Suvaditya's resume and get an answer!",
-        theme="John6666/YntecDark"
+        theme="John6666/YntecDark",
+        save_history=True
     )
-    demo.launch()
+    demo.launch()   
 
     # wandb.finish()