Zamiana na chatbota.

app.py

@@ -26,8 +26,8 @@ with open('vector_map.pkl', 'rb') as f:
     vector_map = pickle.load(f)
 
 # Przykładowe wyszukiwanie
-def szukaj(query_text):
-    top_n_results = find_similar(query_text, vector_map, model, top_n=5)
+def szukaj(query_text, history):
+    top_n_results = find_similar(query_text, vector_map, model, top_n=3)
     context=''
     for text, similarity in top_n_results:
         context=context+text
@@ -37,23 +37,21 @@ def szukaj(query_text):
         max_tokens=512,
         messages=[
             {'role': 'system',
-             'content': 'Nazywasz się Jacek i jesteś ekspertem cyfrowej dostępności. Odpowiadasz konkretnie i krótko na pytania na podstawie kontekstu podanego przez użytkownika. Możesz poprosić o doprecyzowanie, jeżeli nie jesteś pewien kontekstu.'},
+             'content': 'Nazywasz się Jacek Zadrożny i jesteś ekspertem cyfrowej dostępności. Odpowiadasz konkretnie i krótko na pytania. Możesz poprosić o doprecyzowanie, jeżeli nie jesteś pewien kontekstu.'+context},
             {'role': 'user',
-             'content': query_text+context}
+             'content': query_text}
         ]
     )
     return jaczat.choices[0].message.content
-demo=gr.Interface(
+demo=gr.ChatInterface(
     fn=szukaj,
-    inputs=gr.Text(label='Pytanie'),
-    outputs=gr.Markdown(),
     theme=gr.themes.Glass(font='OpenSans'),
+    autofocus=True,
     title='JaCzat',
-    description='Tu możesz zapytać o wszystko dotyczące cyfrowej dostępności, w tym przede wszystkim WCAG. Jeżeli odpowiedź jest Twoim zdaniem błędna lub niesatysfakcjonująca, kliknij na flagę "Źle". A jeżeli odpowiedź jest prawidłowa - kliknij "Dobrze". To pomoże mi rozwijać pomocnika na przyszłość. Nie zbieram żadnych danych osobowych i proszę Cię, żeby ich tu nie wpisywać. Potem będę miał problem z ich usuwaniem.',
+    description='Tu możesz zapytać o wszystko dotyczące cyfrowej dostępności, w tym przede wszystkim WCAG.',
     submit_btn='Zapytaj',
-    clear_btn='Wyczyść',
-    allow_flagging='manual',
+    clear_btn=None,
+    retry_btn=None,
+    undo_btn=None,
     show_progress='minimal',
-    flagging_dir='jaczat.csv',
-    flagging_options=['Dobrze', 'Źle']
 ).launch(show_api=False)