Update app.py

app.py

@@ -151,57 +151,36 @@ def upload_to_google_drive():
         
 
 def generate(prompt, history):
-
-    global isFirstRun,initContext,Name,Occupation,Ethnicity,Gender,Age,context,YearsOfExp
-    
-    #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 and not len(YearsOfExp):
+    global isFirstRun, initContext, Name, Occupation, Ethnicity, Gender, Age, context, YearsOfExp, agreed
     if agreed:
-        firstmsg =""
+        firstmsg = ""
         if isFirstRun:
             context = initContext
             isFirstRun = False
             firstmsg = prompt
-            
-
-        context += """
-                  <|im_start|>nurse
-                  Nurse:"""+prompt+"""
-                  <|im_start|>barry
-                  Barry:
-                  """
-
+        
+        context += f"""
+        Nurse:{prompt}
+        Barry:
+        """
+        
         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
-
-
-      #  for output in llm(input, stream=True, max_tokens=100, ):
-        #    piece = output['choices'][0]['text']
-       #     response += piece
-       #     chatbot[-1] = (chatbot[-1][0], response)
+            response = output["choices"][0]["text"].strip()
 
-       #     yield  response
-            
-        history.append((prompt,response))
-        if  not isFirstRun:
+        history.append((prompt, response))
+        if not isFirstRun:
             chat_log_history.append({"user": prompt, "bot": response})
-            upload_to_google_drive()
         else:
-            chat_log_history.append({"user": firstmsg, "bot": reponse})
-        
+            chat_log_history.append({"user": firstmsg, "bot": response})
+
         context += response
         
-        print (context)
         return "", history
-        
     else:
-        output = "Did you forget to enter your Details? Please go to the User Info Tab and Input your data. "
-        return output
+        output = "Did you forget to enter your Details? Please go to the User Info Tab and Input your data."
+        return output, history
 
 def submit_user_info(name,occupation,yearsofexp,ethnicity,gender,age):
     global Name, Occupation,Ethnicity,Gender,Age,chat_log_name,YearsOfExp