Update app.py

app.py

@@ -145,12 +145,14 @@ def generate(prompt, history):
                   """
 
         response = ""
+        history2 = []
 
         while(len(response) < 1):
             output = llm(context, max_tokens=400, stop=["Nurse:"], echo=False)
             response = output["choices"][0]["text"]
             response = response.strip()
-            yield  response
+            history2.append(("generation", response))
+            yield  history
 
 
       #  for output in llm(input, stream=True, max_tokens=100, ):
@@ -159,12 +161,11 @@ def generate(prompt, history):
        #     chatbot[-1] = (chatbot[-1][0], response)
 
        #     yield  response
-
+            
+        history.append((prompt, response))
         context += response
         print (context)
-
-        history.append(response)
-        yield response
+        yield history
         
     else:
         output = "Did you forget to enter your Details? Please go to the User Info Tab and Input your data. "