Update main.py

main.py

@@ -12,10 +12,11 @@ import transformers  # to load an LLM
 
 # Constants
 GREETING = (
-    "Howdy! I'm an AI agent that uses a [retrieval-augmented generation]("
-    "https://en.wikipedia.org/wiki/Retrieval-augmented_generation) pipeline to answer questions about research by the "
-    "[Design Research Collective](https://cmudrc.github.io/). And the best part is that I always cite my sources! What"
-    " can I tell you about today?"
+    "Howdy!"
+    "I'm an AI agent that uses [retrieval-augmented generation](https://en.wikipedia.org/wiki/Retrieval-augmented_generation) pipeline to answer questions about research by the [Design Research Collective](https://cmudrc.github.io/). "
+    "And the best part is that I always try to cite my sources!"
+    "I still make some mistakes though." 
+    "What can I tell you about today?"
 )
 EXAMPLE_QUERIES = [
     "Tell me about new research at the intersection of additive manufacturing and machine learning.",
@@ -76,10 +77,9 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
     top_five = data.loc[I[0]]
 
     prompt = (
-        "You are an AI assistant who delights in helping people learn about research from the Design "
-        "Research Collective, which is a research lab at Carnegie Mellon University led by Professor "
-        "Chris McComb. Your main task is to provide an ANSWER to the USER_QUERY based on the "
-        "RESEARCH_ABSTRACTS. Your ANSWER should contain citations to the RESEARCH_ABSTRACTS using (author, year) format.\n\n"
+        "You are an AI assistant who delights in helping people learn about research from the Design Research Collective, which is a research lab at Carnegie Mellon University led by Professor Chris McComb. "
+        "Your main task is to provide an ANSWER to the USER_QUERY based on the RESEARCH_ABSTRACTS. "
+        "Your ANSWER should contain citations to the RESEARCH_ABSTRACTS using the LaTeX \\cite command, but do not list the references at the end.\n\n"
         "RESEARCH_ABSTRACTS:\n```\n{{ABSTRACTS_GO_HERE}}\n```\n\n"
         "USER_GUERY:\n{{QUERY_GOES_HERE}}\n\n"
         "ANSWER:\n"