Update app.py

app.py

@@ -133,15 +133,18 @@ def generate(prompt, history):
     
     if not len(Name) == 0 and not len(Occupation) == 0 and not len(Ethnicity) == 0 and not len(Gender) == 0 and not len(Age) == 0:
 
+        firstmsg =""
         if isFirstRun:
             context = initContext
             isFirstRun = False
+            firstmsg = prompt
+            
 
         context += """
                   <|im_start|>nurse
-                  """+prompt+"""
+                  Nurse:"""+prompt+"""
                   <|im_start|>barry
-                  
+                  Barry:
                   """
 
         response = ""
@@ -162,7 +165,14 @@ def generate(prompt, history):
        #     yield  response
             
         history.append((prompt,response))
+        chat_history.append([prompt, response])
+        if  not isFirstRun:
+            chat_log_history.append({"user": prompt, "bot": output})
+            upload_to_google_drive()
+        else:
+            chat_log_history.append({"user": firstmsg, "bot": output})
         context += response
+        
         print (context)
         return history
         
@@ -170,21 +180,6 @@ def generate(prompt, history):
         output = "Did you forget to enter your Details? Please go to the User Info Tab and Input your data. "
         yield output
 
-def predict(input, chatbot, max_length, top_p, temperature, history):
-    chatbot.append((input, ""))
-    response = ""
-    history.append(input)
-
-    for output in llm(input, stream=True, temperature=temperature, top_p=top_p, max_tokens=max_length, ):
-        piece = output['choices'][0]['text']
-        response += piece
-        chatbot[-1] = (chatbot[-1][0], response)
-
-        yield chatbot, history
-
-    history.append(response)
-    yield chatbot, history
-
 
 chat_bot=gr.ChatInterface(
     fn=generate,