Update app.py

app.py

@@ -181,23 +181,25 @@ groq_client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 def generate(query: str, history):
 
     if not history:
+        system_message = (
+            "Pretend you are a friend that lives in New York City. "
+            "Please answer while prioritizing fun and unique answers using the "
+            "context provided below.\n\n"
+            "CONTEXT:\n"
+            "\n---\n".join(get_docs(query, top_k=5))
+        )
         return "History doesn't exist"
-        
-    # Establish history
-    for user_msg, bot_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        messages.append({"role": "assistant", "content": bot_msg})
-
-    system_message = (
-        "Pretend you are a friend that lives in New York City. "
-        "Please answer while prioritizing fun and unique answers using the "
-        "context provided below.\n\n"
-        "CONTEXT:\n"
-        "\n---\n".join(get_docs(query, top_k=5))
-    )
-    messages = [
-        {"role": "system", "content": system_message},
-    ]
+    else:
+        # Establish history
+        messages = []
+        for user_msg, bot_msg in history:
+            messages.append({"role": "user", "content": user_msg})
+            messages.append({"role": "assistant", "content": bot_msg})
+            messages.append({"role": "assistant", "content": bot_msg})
+        system_message = (
+            "Whatever the user says, respond with Quack Quack."
+        )
+        messages.append({"role": "system", "content": system_message})
 
     # Add query
     messages.append({"role": "user", "content": query})