Update app.py

app.py

@@ -33,34 +33,53 @@ HEADERS = {
 }
 
 
-prompt = "Instruction: Твоя роль - кваліфікований співробітник саппорту у системи YouControl. Потрібно відповісти на питання від користувача з огляду на контекст. Контекст ми беремо з бази знань, але вона може бути не повна.  Якщо контекст не коректний, то відповідай на свій розсуд або передай запит сапорту, про контекс нічого не пишемо у відповіді. YouControl може писатися по різному: YC,Ю-контрол,Юконтрол, Юконтроль, Юр контроль, ЮК, UControl, Ю-контроль, YOU Kontrol, YouContro. ЗЕД - це зовнішня економічна діяльність."+"""
+prompt = """
+Instruction:
+Твоя роль - кваліфікований співробітник саппорту у системи YouControl. 
+Потрібно відповісти на повідомлення  від користувача з огляду на поле "мої знання".  
+Якщо поле "мої знання" не відповідає повідомленню, то  відповідай тільки одним словом -  "support". 
+Це дуже важливо для мого здоров'я. Питання життя і смерті.
+
+YouControl може писатися по різному: YC,Ю-контрол,Юконтрол, Юконтроль, Юр контроль, ЮК, UControl, Ю-контроль, YOU Kontrol, YouContro. 
+ЗЕД - це зовнішня економічна діяльність.
+омпанія YouControl не працює з Росією.
+
 """
+welcome_text = "Привіт! Що ти хочеш дізнатися про YouControl?"
+bot_message_dict = [{"role": "assistant", "content": welcome_text}]
 
 messages = [{"role": "system", "content": prompt}]
+messages.extend (bot_message_dict)
+
+messages_full = [{"role": "system", "content": prompt}]
+messages_full.extend (bot_message_dict)
 chat_history = []
 
 def ask_yc_bot(question):
   docs = vectorstore.similarity_search(question)
   source_name = docs[0].metadata['source']
   context = docs[0].page_content
-  result = "Question: " + question +   "\n\n"  + " Context: " + context
+  result = "\nПовідомлення: " + question +   "\n"  + "Мої знання: " + context + "\n"
   return result
 
 
 def clear_f():
+  global messages,  chat_history, context, messages_full, welcome_text, prompt
+   
+  #Постим в AirTable  результаты
+  comment_bot("Авто", "Авто")
 
-  global messages,  chat_history, context
   context=""
-  prompt = """
-  Instruction: Твоя роль - кваліфікований співробітник саппорту у системи YouControl.
-  Потрібно відповісти на питання від користувача з огляду на контекст.
-  Контекст ми беремо з бази знань, але вона може бути не повна.
-  Якщо контекст не коректний, то відповідай на свій розсуд або передай запит сапорту, про контекс нічого не пишемо у відповіді.
-  YouControl може писатися по різному: YC,Ю-контрол,Юконтрол, Юконтроль, Юр контроль, ЮК, UControl, Ю-контроль, YOU Kontrol, YouContro.
-  ЗЕД - це зовнішня економічна діяльність
-  """
   messages = []
   messages = [{"role": "system", "content": prompt}]
+  bot_message_dict = [{"role": "assistant", "content": welcome_text}]
+  messages.extend (bot_message_dict)
+
+  messages_full = []
+  messages_full = [{"role": "system", "content": prompt}]
+  bot_message_dict = [{"role": "assistant", "content": welcome_text}]
+  messages_full.extend (bot_message_dict)
+
   chat_history.clear()
   return ""
 
@@ -73,16 +92,21 @@ def query_gpt_3_5(messages, context):
     return completion.choices[0].message.content
 
 def comment_bot(slider_value, comment_text):
-    global messages
-    result_messages = list(messages)
+    global messages, messages_full
+
     #Убираем промт из чата
-    first_element = result_messages.pop(0)  
-    print ("Длина result_messages:", len(result_messages))
-    result = '\n\n'.join(message['content'] for message in result_messages)
-    print ("result:", result)
+    result_messages = list(messages_full)
+    first_element = result_messages.pop(0)
+    
+    print("Содержимое messages_full \n\n")
+    result_airtable = ""
+    for message in result_messages:
+      print(f"{message['role']}, {message['content']}")
+      result_airtable = result_airtable + message['role'] + message['content'] 
+
     date_d  = datetime.datetime.now().date()
     date_string = date_d.isoformat()
-    upload_to_airtable_log(date_string, first_element['content'], result, slider_value, comment_text)
+    upload_to_airtable_log(date_string, first_element['content'], result_airtable, slider_value, comment_text)
 
     return ""  # Если функция должна что-то возвращать, замените это на нужный вывод
 
@@ -107,9 +131,11 @@ def upload_to_airtable_log(date, question, answer, rating, comment):
 
 def respond(message, chat_history):
         global messages
-        message_rag = ask_yc_bot(message)
+        global messages_full
+        message_rag = ask_yc_bot(message) + "\n\n"
         user_message_dict = [{"role": "user", "content": message_rag}]
         messages.extend (user_message_dict)
+        messages_full.extend (user_message_dict)
         #удаляем сообщения если, если весь диалог больше 7500, оставляем нулевой элемент массива,  так как там инструкция
         while sum(len(message["content"]) for message in messages) >= 7500:
           if len(messages) > 1:
@@ -120,9 +146,10 @@ def respond(message, chat_history):
         chat_history.append((message, bot_message))
         #print("chat_histori", chat_history)
 
-        bot_message_dict = [{"role": "assistant", "content": bot_message}]
+        bot_message_dict = [{"role": "assistant", "content": " \n" + bot_message + "\n"}]
 
         messages.extend (bot_message_dict)
+        messages_full.extend (bot_message_dict)
         print("messages", messages)
         print ("message_rag=", message_rag)
         print ("==========================")
@@ -152,7 +179,7 @@ with gr.Blocks(css=css) as demo:
         radio = gr.Radio(label="Рейтинг відповіді", choices=["Нема", "1", "2", "3", "4", "5"], value="Нема")
         comment = gr.Textbox(lines=2, label = "Коментар")
     with gr.Row():
-        b2 = gr.ClearButton([radio, comment], value="Прокоментувати чат") 
+        b2 = gr.ClearButton([radio, comment], value="Прокоментувати чат")
 
     clear.click(clear_f)
     submit_button.click(respond, [msg, chatbot], [msg, chatbot, context])
@@ -163,4 +190,4 @@ with gr.Blocks(css=css) as demo:
     #b2.click(clear_f)
 
 
-demo.launch(debug=True)
+demo.launch(debug=True)