Update app.py

app.py

@@ -7,20 +7,20 @@ from langchain.vectorstores import Pinecone
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.chains import ConversationalRetrievalChain
 from langchain.chat_models import ChatOpenAI
-import datetime
+from datetime import datetime, timedelta
 import requests
 
 openai.api_key= os.environ.get('API_OPENAI')
-embeddings = OpenAIEmbeddings(openai_api_key=openai.api_key)
-
 PINECONE_API_KEY = os.environ.get('API_PINECONE')
+AIRTABLE_API_KEY = os.environ.get('AIRTABLE_API_KEY')
+
+##########
+embeddings = OpenAIEmbeddings(openai_api_key=openai.api_key)
 pinecone.init(
     api_key=PINECONE_API_KEY,  # find at app.pinecone.io
     environment="eu-west4-gcp"  # next to api key in console
 )
-AIRTABLE_API_KEY = os.environ.get('AIRTABLE_API_KEY')
 
-##########
 
 index_name = "yc-faq-air"
 vectorstore = Pinecone.from_existing_index(index_name, embeddings)
@@ -35,17 +35,15 @@ HEADERS = {
 
 prompt = """
 Instruction:
-Твоя роль - кваліфікований співробітник саппорту у системи YouControl. 
-Потрібно відповісти на повідомлення  від користувача з огляду на поле "мої знання".  
-Якщо поле "мої знання" не відповідає повідомленню, то  відповідай на свій розсуд. 
-Прощатися і дякувати тільки у відповідь.
+Твоя роль - кваліфікований співробітник саппорту у системи YouControl.
+Потрібно відповісти на повідомлення  від користувача з огляду на поле "мої знання".
+Якщо поле "мої знання" не відповідає повідомленню, то  відповідай "support".
+Не вітайся і не прощайся.
 Перевіряв офографію перед відповіддю.
 Це дуже важливо для мого здоров'я. Питання життя і смерті.
-
-YouControl може писатися по різному: YC,Ю-контрол,Юконтрол, Юконтроль, Юр контроль, ЮК, UControl, Ю-контроль, YOU Kontrol, YouContro. 
+YouControl може писатися по різному: YC,Ю-контрол,Юконтрол, Юконтроль, Юр контроль, ЮК, UControl, Ю-контроль, YOU Kontrol, YouContro.
 ЗЕД - це зовнішня економічна діяльність.
 Компанія YouControl не працює з Росією.
-
 """
 welcome_text = "Привіт! Що ти хочеш дізнатися про YouControl?"
 bot_message_dict = [{"role": "assistant", "content": welcome_text}]
@@ -57,17 +55,23 @@ messages_full = [{"role": "system", "content": prompt}]
 messages_full.extend (bot_message_dict)
 chat_history = []
 
-def ask_yc_bot(question):
+def get_current_time():
+    now = datetime.now()
+    two_hours_later = now + timedelta(hours=2)
+    return two_hours_later.strftime('%H:%M:%S')
+
+
+def get_vector(question):
   docs = vectorstore.similarity_search(question)
   source_name = docs[0].metadata['source']
   context = docs[0].page_content
-  result = "\nПовідомлення: " + question +   "\n"  + "Мої знання: " + context + "\n"
+  result =  context
   return result
 
 
 def clear_f():
   global messages,  chat_history, context, messages_full, welcome_text, prompt
-   
+
   #Постим в AirTable  результаты
   comment_bot("Авто", "Авто")
 
@@ -85,12 +89,11 @@ def clear_f():
   chat_history.clear()
   return ""
 
-def query_gpt_3_5(messages, context):
+def query_gpt(messages, context):
     completion = openai.ChatCompletion.create(
         model="gpt-4-0125-preview",
         messages=messages
         )
-    print ("gpt ответил", completion.choices[0].message.content)
     return completion.choices[0].message.content
 
 def comment_bot(slider_value, comment_text):
@@ -99,14 +102,14 @@ def comment_bot(slider_value, comment_text):
     #Убираем промт из чата
     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'] 
+      result_airtable = result_airtable + message['role'] + message['content']
 
-    date_d  = datetime.datetime.now().date()
+    date_d = datetime.now().date()
     date_string = date_d.isoformat()
     upload_to_airtable_log(date_string, first_element['content'], result_airtable, slider_value, comment_text)
 
@@ -132,10 +135,18 @@ def upload_to_airtable_log(date, question, answer, rating, comment):
         print(f"Successfully uploaded airtable log")
 
 def respond(message, chat_history):
+        print (get_current_time())
+        print ("start_respond")
+        print ("message:")
+        print (message)
         global messages
         global messages_full
-        message_rag = ask_yc_bot(message) + "\n\n"
-        user_message_dict = [{"role": "user", "content": message_rag}]
+        message_rag = get_vector(message)
+        print (get_current_time())
+        print ("message_rag:")
+        print (message_rag)
+        print ("=")
+        user_message_dict = [{"role": "system", "content": "Мої знання: " + message_rag}]
         messages.extend (user_message_dict)
         messages_full.extend (user_message_dict)
         #удаляем сообщения если, если весь диалог больше 7500, оставляем нулевой элемент массива,  так как там инструкция
@@ -144,7 +155,10 @@ def respond(message, chat_history):
               del messages[1]
           else:
               break
-        bot_message = query_gpt_3_5(messages, "")
+        bot_message = query_gpt(messages, "")
+        print (get_current_time())
+        print ("GPT answer")
+        print (bot_message)
         chat_history.append((message, bot_message))
         #print("chat_histori", chat_history)
 
@@ -152,8 +166,9 @@ def respond(message, chat_history):
 
         messages.extend (bot_message_dict)
         messages_full.extend (bot_message_dict)
-        print("messages", messages)
-        print ("message_rag=", message_rag)
+        print ("messages")
+        for message in messages:
+           print(f"Роль: {message['role']}, Содержание: '{message['content']}'")
         print ("==========================")
         total_chars = sum(len(message["content"]) for message in messages)
         print(total_chars)
@@ -179,7 +194,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Row():
         clear = gr.ClearButton([msg, chatbot, context, radio, comment], value="Новый чат")
         b2 = gr.ClearButton([radio, comment], value="Прокоментувати чат")
-        
+
 
     clear.click(clear_f)
     submit_button.click(respond, [msg, chatbot], [msg, chatbot, context])
@@ -191,4 +206,4 @@ with gr.Blocks(css=css) as demo:
     #b2.click(clear_f)
 
 
-demo.launch(debug=True)
+demo.launch(debug=True)