Update app.py

app.py

@@ -34,7 +34,7 @@ def get_apartment_data(query):
 
 # Функция для обработки запросов и ответов
 def respond(message, history, system_message, max_tokens, temperature, top_p):
-    print("Получен запрос:", message)  # Логирование для отладки
+    print("Получен запрос:", message)
     print("История общения:", history)
 
     try:
@@ -47,20 +47,18 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
                     f"Класс: {row['project_class']}, Площадь: {row['apt_area']}"
                     for row in response
                 ])
-                history.append((message, result))
-                yield "", history
+                history.append({"role": "user", "content": message})
+                history.append({"role": "assistant", "content": result})
+                yield result, history
             else:
-                history.append((message, "Извините, подходящих вариантов не найдено."))
-                yield "", history
+                history.append({"role": "user", "content": message})
+                history.append({"role": "assistant", "content": "Извините, подходящих вариантов не найдено."})
+                yield "Извините, подходящих вариантов не найдено.", history
         else:
             # Генерация ответа от модели
             messages = [{"role": "system", "content": system_message}]
-            for user_message, bot_response in history:
-                if user_message:
-                    messages.append({"role": "user", "content": user_message})
-                if bot_response:
-                    messages.append({"role": "assistant", "content": bot_response})
-
+            for item in history:
+                messages.append(item)
             messages.append({"role": "user", "content": message})
 
             response = ""
@@ -73,11 +71,12 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
             ):
                 token = message.choices[0].delta.content
                 response += token
-                yield response
+                yield response, history
     except Exception as e:
         print(f"Ошибка обработки запроса: {e}")
-        history.append((message, "Произошла ошибка при обработке вашего запроса."))
-        yield "", history
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": "Произошла ошибка при обработке вашего запроса."})
+        yield "Произошла ошибка при обработке вашего запроса.", history
 
 # Интерфейс Gradio
 demo = gr.ChatInterface(