Update app.py
Browse files
app.py
CHANGED
|
@@ -47,53 +47,80 @@ def model_response(message, history):
|
|
| 47 |
|
| 48 |
message = message["text"]
|
| 49 |
|
|
|
|
|
|
|
|
|
|
| 50 |
# --- Преобразование формата истории Gradio в формат Google GenAI ---
|
| 51 |
gemini_history = []
|
| 52 |
-
# Итерация по истории Gradio.
|
| 53 |
-
# Сделаем проверку на формат элемента.
|
| 54 |
for turn in history:
|
| 55 |
-
# Проверяем, является ли элемент списком/кортежем
|
| 56 |
if isinstance(turn, (list, tuple)) and len(turn) == 2:
|
| 57 |
-
human, assistant = turn #
|
| 58 |
-
|
| 59 |
-
# Добавляем
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
gemini_history.append({'role': 'model', 'parts': [{'text':
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 67 |
else:
|
| 68 |
-
# Если формат
|
| 69 |
-
print(f"Warning: Skipping
|
| 70 |
-
# При необходимости можно сгенерировать ошибку, если такой формат не должен встречаться:
|
| 71 |
-
# raise ValueError(f"Unexpected history format received from Gradio: {turn}")
|
| 72 |
|
| 73 |
|
| 74 |
-
print(f"Отформатированная история для Gemini: {gemini_history}")
|
| 75 |
|
| 76 |
try:
|
|
|
|
|
|
|
|
|
|
| 77 |
# Создать новую сессию чата с историческим контекстом
|
| 78 |
-
#
|
| 79 |
-
|
| 80 |
-
chat = client.chats.create(model=used_model, history=gemini_history)
|
| 81 |
|
| 82 |
# Отправить *текущее* сообщение пользователя
|
| 83 |
-
# Текущее сообщение
|
| 84 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 85 |
|
| 86 |
# Извлечь текстовый ответ
|
| 87 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 88 |
|
| 89 |
print(f"Ответ модели: {response_text}")
|
| 90 |
|
| 91 |
return response_text
|
| 92 |
|
| 93 |
except Exception as e:
|
| 94 |
-
print(f"
|
| 95 |
# Обработка потенциальных ошибок
|
| 96 |
-
|
|
|
|
|
|
|
|
|
|
| 97 |
|
| 98 |
print("Define test1 response work")
|
| 99 |
def random_response(message, history):
|
|
|
|
| 47 |
|
| 48 |
message = message["text"]
|
| 49 |
|
| 50 |
+
print(f"Получено сообщение: {message}")
|
| 51 |
+
print(f"Получена история Gradio: {history}") # Выводим историю для отладки
|
| 52 |
+
|
| 53 |
# --- Преобразование формата истории Gradio в формат Google GenAI ---
|
| 54 |
gemini_history = []
|
| 55 |
+
# Итерация по истории Gradio. Ищем только завершенные пары [пользователь, бот].
|
|
|
|
| 56 |
for turn in history:
|
| 57 |
+
# Проверяем, является ли элемент списком/кортежем из 2 элементов
|
| 58 |
if isinstance(turn, (list, tuple)) and len(turn) == 2:
|
| 59 |
+
human, assistant = turn # Распаковка безопасна
|
| 60 |
+
|
| 61 |
+
# Добавляем ПРЕДЫДУЩИЙ ход только если он ПОЛНОСТЬЮ завершен (есть и пользователь, и бот)
|
| 62 |
+
# И сообщения не пустые после удаления пробелов
|
| 63 |
+
human_text = str(human).strip() if human is not None else ""
|
| 64 |
+
assistant_text = str(assistant).strip() if assistant is not None else ""
|
| 65 |
+
|
| 66 |
+
if human_text != "" and assistant_text != "":
|
| 67 |
+
gemini_history.append({'role': 'user', 'parts': [{'text': human_text}]})
|
| 68 |
+
gemini_history.append({'role': 'model', 'parts': [{'text': assistant_text}]})
|
| 69 |
+
elif human_text != "" and assistant_text == "":
|
| 70 |
+
# Это может быть незавершенный последний ход пользователя.
|
| 71 |
+
# Его не добавляем в history для start_chat, т.к. текущий message уже содержит последний ввод.
|
| 72 |
+
# Можно вывести предупреждение, если это неожиданно
|
| 73 |
+
print(f"Warning: Skipping incomplete user turn in history: {turn}")
|
| 74 |
+
# Если ни human, ни assistant не содержат текста, просто пропускаем.
|
| 75 |
+
# Если assistant есть, а human нет, это тоже странно, пропускаем.
|
| 76 |
+
|
| 77 |
else:
|
| 78 |
+
# Если формат неожиданный (не пара [user, bot]), пропускаем
|
| 79 |
+
print(f"Warning: Skipping non-[user, bot] history entry: {turn}")
|
|
|
|
|
|
|
| 80 |
|
| 81 |
|
| 82 |
+
print(f"Отформатированная история для Gemini (для start_chat): {gemini_history}")
|
| 83 |
|
| 84 |
try:
|
| 85 |
+
# Инициализируем модель здесь, чтобы использовать ее объект
|
| 86 |
+
model = genai.GenerativeModel(used_model)
|
| 87 |
+
|
| 88 |
# Создать новую сессию чата с историческим контекстом
|
| 89 |
+
# Передаем только завершенные предыдущие ходы.
|
| 90 |
+
chat = model.start_chat(history=gemini_history)
|
|
|
|
| 91 |
|
| 92 |
# Отправить *текущее* сообщение пользователя
|
| 93 |
+
# Текущее сообщение 'message' всегда отправляется отдельно методом send_message
|
| 94 |
+
# Убедимся, что текущее сообщение не пустое
|
| 95 |
+
current_message_text = str(message).strip()
|
| 96 |
+
if current_message_text == "":
|
| 97 |
+
return "Пожалуйста, введите ваше сообщение." # Вернуть сообщение, если пользователь отправил пустое
|
| 98 |
+
|
| 99 |
+
response = chat.send_message(current_message_text)
|
| 100 |
|
| 101 |
# Извлечь текстовый ответ
|
| 102 |
+
# Проверим, что ответ не пустой и что он вообще есть
|
| 103 |
+
response_text = ""
|
| 104 |
+
if response and response.text:
|
| 105 |
+
response_text = response.text.strip()
|
| 106 |
+
else:
|
| 107 |
+
# Если ответ пустой или произошла другая проблема с ответом
|
| 108 |
+
print(f"Warning: Received empty or invalid response from model for message: {current_message_text}")
|
| 109 |
+
# Можно попытаться получить дополнительную информацию из response, если доступно
|
| 110 |
+
# print(f"Full response object: {response}")
|
| 111 |
+
return "Извините, я не смог сгенерировать ответ." # Или другое сообщение об ошибке
|
| 112 |
|
| 113 |
print(f"Ответ модели: {response_text}")
|
| 114 |
|
| 115 |
return response_text
|
| 116 |
|
| 117 |
except Exception as e:
|
| 118 |
+
print(f"Error calling Google GenAI API: {e}")
|
| 119 |
# Обработка потенциальных ошибок
|
| 120 |
+
# Можно попробовать получить детали ошибки из API, если они есть в исключении e
|
| 121 |
+
error_message = f"Произошла ошибка при генерации ответа: {e}"
|
| 122 |
+
# TODO: Парсить e, чтобы дать более конкретное сообщение, например, о модерации
|
| 123 |
+
return error_message
|
| 124 |
|
| 125 |
print("Define test1 response work")
|
| 126 |
def random_response(message, history):
|