Update app.py

app.py

@@ -17,13 +17,13 @@ def generate_text(input_text, history):
         input_text_with_history = f"Q: {input_text} \n A:"
     else:
         input_text_with_history = ""
-        for i in history:
+        for i in history[-1]:
             input_text_with_history += i[0] + "\n" + i[1]
         input_text_with_history += f"Q: {input_text} \n A:"
         
     print("new input", input_text_with_history)
     output = llm(input_text_with_history, max_tokens=1024, stop=["Q:", "\n"], echo=True)
-    response = output['choices'][0]['text']
+    response = output['choices'][0]['text'] + "\n"
     return response