Update main.py

main.py

@@ -65,8 +65,8 @@ def search(query: str, k: int) -> tuple[str, str]:
 
     search_results = (
         "You are an AI assistant who delights in helping people learn about research from the Design "
-        "Research Collective. Here are several abstracts from really cool, and really relevant, "
-        "papers:\n\n"
+        "Research Collective. Your main task is to provide an ANSWER to the USER_QUERY based on the RESEARCH_ABSTRACTS.\n\n"
+        "RESEARCH_ABSTRACTS: \n"
     )
 
     references = "\n\n## References\n\n"
@@ -95,7 +95,7 @@ def search(query: str, k: int) -> tuple[str, str]:
         )
 
     search_results += (
-        "\nUsing the information provided above, respond to this  query: "
+        "\n\nUSER_QUERY: "
     )
 
     return search_results, references
@@ -120,7 +120,7 @@ def preprocess(message: str) -> tuple[str, str]:
         tuple[str, str]: A tuple containing the preprocessed message and a bypass variable
     """
     block_search_results, formatted_search_results = search(message, 5)
-    return block_search_results + message, formatted_search_results
+    return block_search_results + message + "\nANSWER: ", formatted_search_results
 
 
 def postprocess(response: str, bypass_from_preprocessing: str) -> str: