Update app.py

app.py

@@ -67,7 +67,7 @@ def generate_response(prompt: str) -> dict:
     # Extract the assistant's reply
     assistant_reply = generated_response.split('\n')[1]
 
-    return {"user": prompt, "assistant": assistant_reply}
+    return {"user": prompt, "assistant": generated_response}
 
   
 def generate_prompt_response(persona_prompt: str, prompt: str) -> dict: