Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,20 +1,9 @@
|
|
1 |
-
import asyncio
|
2 |
-
import sys
|
3 |
-
|
4 |
-
if sys.platform.startswith('win'):
|
5 |
-
# Используем более стабильную политику событий для Windows
|
6 |
-
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
7 |
-
|
8 |
-
# Создаем новый цикл событий
|
9 |
-
loop = asyncio.new_event_loop()
|
10 |
-
asyncio.set_event_loop(loop)
|
11 |
-
|
12 |
import g4f
|
13 |
import gradio as gr
|
14 |
import tempfile
|
15 |
import os
|
16 |
from pathlib import Path
|
17 |
-
from g4f.client import Client
|
18 |
from PIL import Image
|
19 |
import io
|
20 |
import requests
|
@@ -25,7 +14,7 @@ import sys
|
|
25 |
import time
|
26 |
from typing import Dict, Any, Optional
|
27 |
import easyocr
|
28 |
-
import
|
29 |
|
30 |
AVAILABLE_MODELS = {
|
31 |
"GPT-4": "gpt-4",
|
@@ -37,16 +26,19 @@ AVAILABLE_MODELS = {
|
|
37 |
SYSTEM_PROMPT = """
|
38 |
1. Всегда используйте кодировку UTF-8
|
39 |
2. Используйте обработку ошибок
|
40 |
-
3. Если пользователь отправляет изображение или пишет
|
41 |
- Автоматически проанализируйте изображение
|
42 |
- Опишите что вы видите на русском языке
|
43 |
- Тебе дается текстовая информация, которая может быть полезна для анализа
|
44 |
-
4. Если пользователь просит
|
|
|
|
|
|
|
45 |
- Начните ответ с [GENERATE_IMAGE]
|
46 |
-
- Напишите детальный промпт на
|
47 |
- Завершите промпт строкой [/GENERATE_IMAGE]
|
48 |
- Продолжите обычный ответ на русском языке
|
49 |
-
|
50 |
- Используйте [THINKING_STEP] для обозначения шага размышления
|
51 |
- Используйте [FINAL_ANSWER] для итогового ответа
|
52 |
- Старайтесь подробно описывать ход мыслей
|
@@ -102,7 +94,7 @@ def process_file(file):
|
|
102 |
return None
|
103 |
|
104 |
try:
|
105 |
-
# Получаем расширение
|
106 |
file_extension = file.name.lower().split('.')[-1] if hasattr(file, 'name') else ''
|
107 |
|
108 |
# Список поддерживаемых форматов изображений
|
@@ -154,11 +146,34 @@ def generate_image(prompt: str) -> str:
|
|
154 |
try:
|
155 |
client = Client()
|
156 |
response = client.images.generate(
|
157 |
-
model="flux",
|
158 |
prompt=prompt,
|
159 |
-
response_format="url"
|
|
|
160 |
)
|
161 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
162 |
except Exception as e:
|
163 |
return f"Ошибка при генерации изобраения: {str(e)}"
|
164 |
|
@@ -211,7 +226,7 @@ def process_image(image_input) -> str:
|
|
211 |
if not detected_objects and not detected_text:
|
212 |
return """
|
213 |
<div class="image-analysis-animation">
|
214 |
-
<div class="analysis-text">К сожалению, я не смог распознать объекты на этом
|
215 |
</div>
|
216 |
"""
|
217 |
|
@@ -256,105 +271,291 @@ def process_image(image_input) -> str:
|
|
256 |
if 'tmp' in locals():
|
257 |
Path(tmp.name).unlink(missing_ok=True)
|
258 |
|
259 |
-
|
260 |
messages = [{"role": "system", "content": SYSTEM_PROMPT}]
|
261 |
history = history or []
|
262 |
history.append({"role": "user", "content": message})
|
263 |
history.append({"role": "assistant", "content": ""})
|
264 |
-
|
265 |
try:
|
266 |
-
|
267 |
-
|
268 |
-
|
269 |
-
|
270 |
-
|
|
|
271 |
|
272 |
-
|
273 |
-
|
274 |
-
|
275 |
-
|
276 |
-
|
277 |
-
|
278 |
-
5. Обязательно сделай {num_steps} шагов, не больше и не меньше
|
279 |
|
280 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
281 |
|
282 |
-
|
283 |
-
|
|
|
|
|
|
|
284 |
|
285 |
-
|
286 |
-
<div class="thinking-animation">
|
287 |
-
<div class="thinking-step">🤔 Начинаю анализ (глубина: {thinking_depth:.1f}, шагов: {num_steps})</div>
|
288 |
-
</div>
|
289 |
-
"""
|
290 |
-
history[-1]["content"] = thinking_html
|
291 |
yield history, *[gr.update(visible=False) for _ in range(6)]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
292 |
|
293 |
-
|
294 |
-
|
295 |
-
|
296 |
-
|
297 |
-
|
298 |
-
|
299 |
-
|
300 |
-
|
301 |
-
|
302 |
-
|
303 |
-
|
304 |
-
|
305 |
-
|
306 |
-
|
307 |
-
|
308 |
-
|
309 |
-
|
310 |
-
|
311 |
-
|
312 |
-
|
313 |
-
|
314 |
-
|
315 |
-
|
316 |
-
|
317 |
-
|
318 |
-
|
319 |
-
|
320 |
-
|
321 |
-
|
322 |
-
|
323 |
-
|
324 |
-
|
325 |
-
|
326 |
-
|
327 |
-
|
328 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
329 |
|
330 |
-
|
331 |
-
|
332 |
-
|
333 |
-
|
334 |
-
|
335 |
-
|
336 |
-
|
337 |
-
|
338 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
339 |
yield history, *[gr.update(visible=False) for _ in range(6)]
|
340 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
341 |
|
342 |
-
except Exception as e:
|
343 |
-
history[-1]["content"] = f"❌ Ошибка при анализе: {str(e)}"
|
344 |
-
yield history, *[gr.update(visible=False) for _ in range(6)]
|
345 |
-
# finally:
|
346 |
-
# await client.close()
|
347 |
-
|
348 |
-
return
|
349 |
-
|
350 |
else:
|
351 |
# Обычный режим работы
|
352 |
if direct_mode:
|
353 |
# Прямая отправка в нейронку для генерации изображения
|
354 |
image_url = generate_image(message)
|
355 |
if not image_url.startswith("Ошибка"):
|
356 |
-
|
357 |
-
history[-1]["content"] = f"\n\n{image_analysis}"
|
358 |
else:
|
359 |
history[-1]["content"] = f"❌ {image_url}"
|
360 |
yield history, *[gr.update(visible=False) for _ in range(6)]
|
@@ -376,7 +577,7 @@ async def chat_response(message, history, model_name, direct_mode, thinking_dept
|
|
376 |
if file_content:
|
377 |
message = f"Файл содержит:\n```\n{file_content}\n```\n\n{message}"
|
378 |
|
379 |
-
#
|
380 |
for msg in history[:-2]:
|
381 |
messages.append({"role": msg["role"], "content": msg["content"]})
|
382 |
messages.append({"role": "user", "content": str(message)})
|
@@ -401,22 +602,22 @@ async def chat_response(message, history, model_name, direct_mode, thinking_dept
|
|
401 |
end_idx = partial_message.find("[/GENERATE_IMAGE]")
|
402 |
image_prompt = partial_message[start_idx:end_idx].strip()
|
403 |
|
404 |
-
# Показываем статус
|
405 |
history[-1]["content"] = """
|
406 |
-
<div class="generating-animation">
|
407 |
-
|
408 |
-
</div>
|
409 |
-
"""
|
410 |
yield history, *[gr.update(visible=False) for _ in range(6)]
|
411 |
|
412 |
# Генерируем изображение
|
413 |
image_url = generate_image(image_prompt)
|
414 |
|
415 |
if not image_url.startswith("Ошибка"):
|
416 |
-
# Анализируем изображение
|
417 |
-
image_analysis = process_image(image_url)
|
418 |
explanation_text = partial_message[end_idx + len("[/GENERATE_IMAGE]"):].strip()
|
419 |
-
partial_message = f"\n\n{
|
420 |
else:
|
421 |
partial_message = f"❌ {image_url}"
|
422 |
|
@@ -430,9 +631,13 @@ async def chat_response(message, history, model_name, direct_mode, thinking_dept
|
|
430 |
if code_end != -1:
|
431 |
code_block = partial_message[code_start:code_end].strip()
|
432 |
|
433 |
-
# В конце
|
434 |
history[-1]["content"] = partial_message
|
435 |
-
yield
|
|
|
|
|
|
|
|
|
436 |
|
437 |
except Exception as e:
|
438 |
print(f"Error: {e}")
|
@@ -446,7 +651,7 @@ async def chat_response(message, history, model_name, direct_mode, thinking_dept
|
|
446 |
gr.update(visible=True),
|
447 |
"",
|
448 |
gr.update(visible=False),
|
449 |
-
gr.update(
|
450 |
)
|
451 |
|
452 |
def analyze_code(code):
|
@@ -476,7 +681,7 @@ def analyze_code(code):
|
|
476 |
return [{"role": "assistant", "content": f"❌ Ошибка при анализе: {str(e)}"}]
|
477 |
|
478 |
def create_interface():
|
479 |
-
# Обновляем тему на
|
480 |
theme = gr.themes.Soft(
|
481 |
primary_hue="slate",
|
482 |
secondary_hue="gray",
|
@@ -491,7 +696,7 @@ def create_interface():
|
|
491 |
input_background_fill="#2d2d2d",
|
492 |
)
|
493 |
|
494 |
-
# Обновляем CSS стили для
|
495 |
css = """
|
496 |
.gradio-container {
|
497 |
background-color: #1a1a1a !important;
|
@@ -513,7 +718,7 @@ def create_interface():
|
|
513 |
position: relative;
|
514 |
transition: all 0.3s ease;
|
515 |
}
|
516 |
-
/* Добавляем стиль
|
517 |
.message.bot pre {
|
518 |
background-color: #1a1a1a !important;
|
519 |
border-radius: 8px;
|
@@ -745,6 +950,69 @@ def create_interface():
|
|
745 |
margin-bottom: 10px;
|
746 |
font-style: italic;
|
747 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
748 |
"""
|
749 |
|
750 |
# Добавте этот скрипт в create_interface()
|
@@ -768,7 +1036,7 @@ def create_interface():
|
|
768 |
render_markdown=True
|
769 |
)
|
770 |
|
771 |
-
# Добавляем
|
772 |
output_text = gr.Textbox(
|
773 |
label="Вывод",
|
774 |
interactive=False,
|
@@ -810,16 +1078,16 @@ def create_interface():
|
|
810 |
|
811 |
# Добавляем слайдер глубины размышления (изначально скрытый)
|
812 |
thinking_depth = gr.Slider(
|
813 |
-
minimum=0
|
814 |
-
maximum=
|
815 |
-
value=
|
816 |
-
step=0
|
817 |
-
label="Глубина размышления",
|
818 |
visible=False
|
819 |
)
|
820 |
|
821 |
direct_mode = gr.Checkbox(
|
822 |
-
label="Прямая
|
823 |
value=False
|
824 |
)
|
825 |
file_output = gr.File(label="Загрузить файл")
|
@@ -931,12 +1199,17 @@ def create_interface():
|
|
931 |
|
932 |
if __name__ == "__main__":
|
933 |
demo = create_interface()
|
|
|
|
|
|
|
|
|
934 |
|
935 |
-
|
936 |
-
|
937 |
-
|
938 |
-
|
939 |
-
|
940 |
-
|
941 |
-
|
942 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import g4f
|
2 |
import gradio as gr
|
3 |
import tempfile
|
4 |
import os
|
5 |
from pathlib import Path
|
6 |
+
from g4f.client import Client
|
7 |
from PIL import Image
|
8 |
import io
|
9 |
import requests
|
|
|
14 |
import time
|
15 |
from typing import Dict, Any, Optional
|
16 |
import easyocr
|
17 |
+
from web_engine import search_info, get_page_text # Импортируем функции из web_engine.py
|
18 |
|
19 |
AVAILABLE_MODELS = {
|
20 |
"GPT-4": "gpt-4",
|
|
|
26 |
SYSTEM_PROMPT = """
|
27 |
1. Всегда используйте кодировку UTF-8
|
28 |
2. Используйте обработку ошибок
|
29 |
+
3. Если пользователь отправляет изображение или пишет что-то на подобие 'посмотри на изображение':
|
30 |
- Автоматически проанализируйте изображение
|
31 |
- Опишите что вы видите на русском языке
|
32 |
- Тебе дается текстовая информация, которая может быть полезна для анализа
|
33 |
+
4. Если пользователь просит найти информацию, например, "найди что-то про Адольфа Гитлера":
|
34 |
+
- Ответь с текстом [SEARCH: {prompt}]
|
35 |
+
- Затем используй полученную информацию для ответа на вопрос
|
36 |
+
5. Если пользователь просит нарисовать или сгенерировать изображение:
|
37 |
- Начните ответ с [GENERATE_IMAGE]
|
38 |
+
- Напишите детальный промпт на английском языке
|
39 |
- Завершите промпт строкой [/GENERATE_IMAGE]
|
40 |
- Продолжите обычный ответ на русском языке
|
41 |
+
6. В режиме размышления:
|
42 |
- Используйте [THINKING_STEP] для обозначения шага размышления
|
43 |
- Используйте [FINAL_ANSWER] для итогового ответа
|
44 |
- Старайтесь подробно описывать ход мыслей
|
|
|
94 |
return None
|
95 |
|
96 |
try:
|
97 |
+
# Получаем расширение файла
|
98 |
file_extension = file.name.lower().split('.')[-1] if hasattr(file, 'name') else ''
|
99 |
|
100 |
# Список поддерживаемых форматов изображений
|
|
|
146 |
try:
|
147 |
client = Client()
|
148 |
response = client.images.generate(
|
149 |
+
model="flux-1.1-pro",
|
150 |
prompt=prompt,
|
151 |
+
response_format="url",
|
152 |
+
provider=g4f.Provider.Airforce
|
153 |
)
|
154 |
+
|
155 |
+
image_url = response.data[0].url
|
156 |
+
|
157 |
+
# Проверяем доступность изображения
|
158 |
+
timeout = 30 # Максимальное время ожидания в секундах
|
159 |
+
interval = 0.1 # Интервал между проверками в секундах
|
160 |
+
elapsed_time = 0
|
161 |
+
|
162 |
+
while elapsed_time < timeout:
|
163 |
+
# Проверяем доступность изображения
|
164 |
+
head_response = requests.head(image_url)
|
165 |
+
if head_response.status_code == 200:
|
166 |
+
# Дополнительная проверка на наличие изображения
|
167 |
+
get_response = requests.get(image_url)
|
168 |
+
if get_response.status_code == 200:
|
169 |
+
return image_url # Изображение доступно
|
170 |
+
else:
|
171 |
+
return "❌ Ошибка: изображение недоступно."
|
172 |
+
time.sleep(interval)
|
173 |
+
elapsed_time += interval
|
174 |
+
|
175 |
+
return "❌ Ошибка: время ожидания изображения истекло." # Время ожидания истекло
|
176 |
+
|
177 |
except Exception as e:
|
178 |
return f"Ошибка при генерации изобраения: {str(e)}"
|
179 |
|
|
|
226 |
if not detected_objects and not detected_text:
|
227 |
return """
|
228 |
<div class="image-analysis-animation">
|
229 |
+
<div class="analysis-text">К сожалению, я не смог распознать объекты на этом изобраении</div>
|
230 |
</div>
|
231 |
"""
|
232 |
|
|
|
271 |
if 'tmp' in locals():
|
272 |
Path(tmp.name).unlink(missing_ok=True)
|
273 |
|
274 |
+
def chat_response(message, history, model_name, direct_mode, thinking_depth=1.0, uploaded_file=None):
|
275 |
messages = [{"role": "system", "content": SYSTEM_PROMPT}]
|
276 |
history = history or []
|
277 |
history.append({"role": "user", "content": message})
|
278 |
history.append({"role": "assistant", "content": ""})
|
279 |
+
|
280 |
try:
|
281 |
+
# Проверяем, содержит ли сообщение [SEARCH:]
|
282 |
+
if "[SEARCH:" in message:
|
283 |
+
print("Поиск информации...") # Отладочное сообщение
|
284 |
+
prompt = message.split("[SEARCH:")[1].strip(" ]") # Извлекаем текст запроса
|
285 |
+
response_text = f"[SEARCH: {prompt}]\n" # Форматируем ответ
|
286 |
+
history[-1]["content"] = response_text # Обновляем историю с запросом поиска
|
287 |
|
288 |
+
# Выполняем поиск информации
|
289 |
+
results = search_info(prompt) # Ищем информацию
|
290 |
+
if not results:
|
291 |
+
history.append({"role": "assistant", "content": "Не удалось найти информацию."})
|
292 |
+
yield history, *[gr.update(visible=False) for _ in range(6)]
|
293 |
+
return
|
|
|
294 |
|
295 |
+
info_text = ""
|
296 |
+
for link in results:
|
297 |
+
info_text += f"Ссылка: {link}\n"
|
298 |
+
text = get_page_text(link)
|
299 |
+
info_text += f"Текст: {text}\n\n" # Добавляем текст страницы
|
300 |
+
print(f"Вот информация по запросу '{prompt}':\n\n{info_text}\nСделай краткий ответ на основе этой информации.")
|
301 |
+
# Формируем запрос для GPT
|
302 |
+
web_search = f"Вот информация по запросу '{prompt}':\n\n{info_text}\nСделай краткий ответ на основе этой информации."
|
303 |
+
print("Отправка запроса к GPT...") # Отладочное сообщение
|
304 |
+
response = g4f.ChatCompletion.create(
|
305 |
+
model="gpt-4o",
|
306 |
+
messages=[
|
307 |
+
{"role": "system", "content": "Сделай краткий ответ на основе следующей информации:"},
|
308 |
+
{"role": "user", "content": web_search}
|
309 |
+
],
|
310 |
+
stream=True,
|
311 |
+
)
|
312 |
|
313 |
+
# Получаем ответ от GPT
|
314 |
+
if isinstance(response, str):
|
315 |
+
final_response = response
|
316 |
+
else:
|
317 |
+
final_response = "Не удалось получить ответ."
|
318 |
|
319 |
+
history.append({"role": "assistant", "content": final_response}) # Добавляе�� ответ ИИ
|
|
|
|
|
|
|
|
|
|
|
320 |
yield history, *[gr.update(visible=False) for _ in range(6)]
|
321 |
+
return
|
322 |
+
|
323 |
+
if model_name == "GPT-4o THINK":
|
324 |
+
max_retries = 5
|
325 |
+
retry_count = 0
|
326 |
+
|
327 |
+
# Формируем thinking_prompt
|
328 |
+
min_steps = 2
|
329 |
+
max_steps = 25
|
330 |
+
thinking_depth = float(thinking_depth)
|
331 |
+
num_steps = max(min_steps, int(thinking_depth))
|
332 |
+
|
333 |
+
thinking_prompt = f"""Ты - система глубокого анализа. Твоя задача - максимально подробно проанализировать вопрос.
|
334 |
|
335 |
+
Глубина анализа: {thinking_depth:.1f}/10
|
336 |
+
Количество шагов размышления: {num_steps}
|
337 |
+
|
338 |
+
ВАЖНО: Строго соблюдай формат ответа. Каждый тег должен быть на новой строке:
|
339 |
+
|
340 |
+
[THINKING_STEP]
|
341 |
+
Текст первого шага...
|
342 |
+
[/THINKING_STEP]
|
343 |
+
|
344 |
+
[THINKING_STEP]
|
345 |
+
Текст второго шага...
|
346 |
+
[/THINKING_STEP]
|
347 |
+
|
348 |
+
[FINAL_ANSWER]
|
349 |
+
Итоговый ответ...
|
350 |
+
[/FINAL_ANSWER]
|
351 |
+
|
352 |
+
Вопрос: {message}
|
353 |
+
"""
|
354 |
+
|
355 |
+
while retry_count < max_retries:
|
356 |
+
try:
|
357 |
+
# Показываем статус попытки
|
358 |
+
thinking_html = f"""
|
359 |
+
<div class="thinking-animation">
|
360 |
+
<div class="thinking-step">🤔 Думаю...</div>
|
361 |
+
</div>
|
362 |
+
"""
|
363 |
+
history[-1]["content"] = thinking_html
|
364 |
+
yield history, *[gr.update(visible=False) for _ in range(6)]
|
365 |
+
print('Обработка запроса')
|
366 |
+
# Создаем запрос с таймаутом только для первого чанка
|
367 |
+
response = g4f.ChatCompletion.create(
|
368 |
+
model="gpt-4o",
|
369 |
+
messages=[{"role": "system", "content": thinking_prompt}],
|
370 |
+
stream=True,
|
371 |
+
# timeout=30
|
372 |
+
)
|
373 |
+
|
374 |
+
# Флаг для отслеживания получения первого чанка
|
375 |
+
received_first_chunk = False
|
376 |
+
partial_message = ""
|
377 |
+
current_block = ""
|
378 |
+
in_thinking_step = False
|
379 |
+
in_final_answer = False
|
380 |
+
|
381 |
+
for chunk in response:
|
382 |
+
if chunk and isinstance(chunk, str):
|
383 |
+
received_first_chunk = True
|
384 |
+
current_block += chunk
|
385 |
+
|
386 |
+
# Исправляем неправильно закрытые теги
|
387 |
+
# ... существующий код ...
|
388 |
+
current_block = (current_block
|
389 |
+
.replace("[/THINKING_STEP\n", "[/THINKING_STEP]\n") # Исправляем закрывающий тег THINKING_STEP
|
390 |
+
.replace("[/FINAL_ANSWER\n", "[/FINAL_ANSWER]\n") # Исправляем закрывающий тег FINAL_ANSWER
|
391 |
+
.replace("\n[/THINKING_STEP", "\n[/THINKING_STEP]") # Исправляем порядок закрывающего тега THINKING_STEP
|
392 |
+
.replace("\n[/FINAL_ANSWER", "\n[/FINAL_ANSWER]") # Исправляем порядок закрывающего тега FINAL_ANSWER
|
393 |
+
.replace("]]]]]]", "") # Убираем лишние закрывающие скобки
|
394 |
+
.replace("]]", "") # Убираем лишние закрывающие скобки
|
395 |
+
.replace("]", "") # Исправляем закрывающий тег THINKING_STEP с учетом закрывающей скобки
|
396 |
+
)
|
397 |
+
# ... существующий код ...
|
398 |
+
|
399 |
+
# Проверяем начало блоков
|
400 |
+
if "[THINKING_STEP]" in current_block:
|
401 |
+
in_thinking_step = True
|
402 |
+
current_block = current_block.replace("[THINKING_STEP]", "")
|
403 |
+
|
404 |
+
if "[THINKING_STEP" in current_block:
|
405 |
+
in_thinking_step = True
|
406 |
+
current_block = current_block.replace("[THINKING_STEP", "")
|
407 |
+
|
408 |
+
if "[FINAL_ANSWER]" in current_block:
|
409 |
+
in_final_answer = True
|
410 |
+
current_block = current_block.replace("[FINAL_ANSWER]", "")
|
411 |
+
|
412 |
+
# Проверяем завершение блоков
|
413 |
+
if "[/THINKING_STEP]" in current_block and in_thinking_step:
|
414 |
+
block_content = current_block[:current_block.find("[/THINKING_STEP]")]
|
415 |
+
formatted_block = f'<div class="thinking-step-block">{block_content}</div>'
|
416 |
+
partial_message += formatted_block
|
417 |
+
current_block = current_block[current_block.find("[/THINKING_STEP]") + len("[/THINKING_STEP]"):]
|
418 |
+
in_thinking_step = False
|
419 |
+
|
420 |
+
if "[/THINKING_STEP" in current_block and in_thinking_step:
|
421 |
+
block_content = current_block[:current_block.find("[/THINKING_STEP")]
|
422 |
+
formatted_block = f'<div class="thinking-step-block">{block_content}</div>'
|
423 |
+
partial_message += formatted_block
|
424 |
+
current_block = current_block[current_block.find("[/THINKING_STEP") + len("[/THINKING_STEP"):]
|
425 |
+
in_thinking_step = False
|
426 |
+
|
427 |
+
if "[/FINAL_ANSWER]" in current_block and in_final_answer:
|
428 |
+
block_content = current_block[:current_block.find("[/FINAL_ANSWER]")]
|
429 |
+
formatted_block = f'<div class="final-answer-block">{block_content}</div>'
|
430 |
+
partial_message += formatted_block
|
431 |
+
current_block = current_block[current_block.find("[/FINAL_ANSWER]") + len("[/FINAL_ANSWER]"):]
|
432 |
+
in_final_answer = False
|
433 |
+
|
434 |
+
if "[/FINAL_ANSWER" in current_block and in_final_answer:
|
435 |
+
block_content = current_block[:current_block.find("[/FINAL_ANSWER")]
|
436 |
+
formatted_block = f'<div class="final-answer-block">{block_content}</div>'
|
437 |
+
partial_message += formatted_block
|
438 |
+
current_block = current_block[current_block.find("[/FINAL_ANSWER") + len("[/FINAL_ANSWER"):]
|
439 |
+
in_final_answer = False
|
440 |
+
|
441 |
+
# Убираем лишние пробелы и переносы строк
|
442 |
+
display_message = partial_message + current_block.strip()
|
443 |
+
history[-1]["content"] = display_message
|
444 |
+
yield history, *[gr.update(visible=False) for _ in range(6)]
|
445 |
+
|
446 |
+
# # Если получили первый чанк, создаем новый стрим без таймаута
|
447 |
+
# if received_first_chunk:
|
448 |
+
# response = g4f.ChatCompletion.create(
|
449 |
+
# model="gpt-4o",
|
450 |
+
# messages=[{"role": "system", "content": thinking_prompt}],
|
451 |
+
# stream=True,
|
452 |
+
# provider=g4f.Provider.Airforce
|
453 |
+
# )
|
454 |
+
# break
|
455 |
+
|
456 |
+
# Продолжаем обработку без таймаута
|
457 |
+
if received_first_chunk:
|
458 |
+
for chunk in response:
|
459 |
+
if chunk and isinstance(chunk, str):
|
460 |
+
current_block += chunk
|
461 |
+
|
462 |
+
# Исправляем неправильно закрытые теги
|
463 |
+
# ... существующий код ...
|
464 |
+
current_block = (current_block
|
465 |
+
.replace("[/THINKING_STEP\n", "[/THINKING_STEP]\n") # Исправляем закрывающий тег THINKING_STEP
|
466 |
+
.replace("[/THINKING_STEP", "[/THINKING_STEP]\n") # Исправляем закрывающий тег THINKING_STEP
|
467 |
+
# .replace("[/THINKING_STEP]", "[/THINKING_STEP]\n") # Исправляем закрывающий тег THINKING_STEP с учетом закрывающей скобки
|
468 |
+
|
469 |
+
.replace("[/FINAL_ANSWER\n", "[/FINAL_ANSWER]\n") # Исправляем закрывающий тег FINAL_ANSWER
|
470 |
+
.replace("[/FINAL_ANSWER]", "[/FINAL_ANSWER]\n") # Исправляем закрывающий тег FINAL_ANSWER с учетом закрывающей скобки
|
471 |
+
.replace("\n[/THINKING_STEP", "\n[/THINKING_STEP]") # Исправляем порядок закрывающего тега THINKING_STEP
|
472 |
+
.replace("\n[/FINAL_ANSWER", "\n[/FINAL_ANSWER]") # Исправляем порядок закрывающего тега FINAL_ANSWER
|
473 |
+
.replace("]]]]]]", "") # Убираем лишние закрывающие скобки
|
474 |
+
.replace("]]", "") # Убираем лишние закрывающие скобки
|
475 |
+
.replace("]", "") # Убираем лишние закрывающие скобки
|
476 |
+
|
477 |
+
)
|
478 |
+
|
479 |
|
480 |
+
# Проверяем начало блоков
|
481 |
+
if "[THINKING_STEP]" in current_block:
|
482 |
+
in_thinking_step = True
|
483 |
+
current_block = current_block.replace("[THINKING_STEP]", "")
|
484 |
+
if "[THINKING_STEP" in current_block:
|
485 |
+
in_thinking_step = True
|
486 |
+
current_block = current_block.replace("[THINKING_STEP", "")
|
487 |
+
|
488 |
+
if "[FINAL_ANSWER]" in current_block:
|
489 |
+
in_final_answer = True
|
490 |
+
current_block = current_block.replace("[FINAL_ANSWER]", "")
|
491 |
+
if "[FINAL_ANSWER" in current_block:
|
492 |
+
in_final_answer = True
|
493 |
+
current_block = current_block.replace("[FINAL_ANSWER", "")
|
494 |
+
|
495 |
+
# Проверяем завершение блоков
|
496 |
+
if "[/THINKING_STEP]" in current_block and in_thinking_step:
|
497 |
+
block_content = current_block[:current_block.find("[/THINKING_STEP]")]
|
498 |
+
formatted_block = f'<div class="thinking-step-block">{block_content}</div>'
|
499 |
+
partial_message += formatted_block
|
500 |
+
current_block = current_block[current_block.find("[/THINKING_STEP]") + len("[/THINKING_STEP]"):]
|
501 |
+
in_thinking_step = False
|
502 |
+
if "[/THINKING_STEP" in current_block and in_thinking_step:
|
503 |
+
block_content = current_block[:current_block.find("[/THINKING_STEP")]
|
504 |
+
formatted_block = f'<div class="thinking-step-block">{block_content}</div>'
|
505 |
+
partial_message += formatted_block
|
506 |
+
current_block = current_block[current_block.find("[/THINKING_STEP") + len("[/THINKING_STEP"):]
|
507 |
+
in_thinking_step = False
|
508 |
+
|
509 |
+
elif "[/FINAL_ANSWER]" in current_block and in_final_answer:
|
510 |
+
block_content = current_block[:current_block.find("[/FINAL_ANSWER]")]
|
511 |
+
formatted_block = f'<div class="final-answer-block">{block_content}</div>'
|
512 |
+
partial_message += formatted_block
|
513 |
+
current_block = current_block[current_block.find("[/FINAL_ANSWER") + len("[/FINAL_ANSWER]"):]
|
514 |
+
in_final_answer = False
|
515 |
+
elif "[/FINAL_ANSWER" in current_block and in_final_answer:
|
516 |
+
block_content = current_block[:current_block.find("[/FINAL_ANSWER")]
|
517 |
+
formatted_block = f'<div class="final-answer-block">{block_content}</div>'
|
518 |
+
partial_message += formatted_block
|
519 |
+
current_block = current_block[current_block.find("[/FINAL_ANSWER") + len("[/FINAL_ANSWER"):]
|
520 |
+
in_final_answer = False
|
521 |
+
|
522 |
+
# Убираем лишние пробелы и переносы строк
|
523 |
+
display_message = partial_message + current_block.strip()
|
524 |
+
history[-1]["content"] = display_message
|
525 |
yield history, *[gr.update(visible=False) for _ in range(6)]
|
526 |
+
break
|
527 |
+
|
528 |
+
retry_count += 1
|
529 |
+
if retry_count == max_retries:
|
530 |
+
error_message = f"""
|
531 |
+
<div class="error-animation">
|
532 |
+
<div class="error-text">❌ Не удалось получить ответ после {max_retries} попыток.
|
533 |
+
Попробуйте использовать обычный режим или повторите запрос позже.</div>
|
534 |
+
</div>
|
535 |
+
"""
|
536 |
+
history[-1]["content"] = error_message
|
537 |
+
yield history, *[gr.update(visible=False) for _ in range(6)]
|
538 |
+
return
|
539 |
+
|
540 |
+
except Exception as e:
|
541 |
+
retry_count += 1
|
542 |
+
if retry_count == max_retries:
|
543 |
+
error_message = f"""
|
544 |
+
<div class="error-animation">
|
545 |
+
<div class="error-text">❌ Ошибка при анализе: {str(e)}</div>
|
546 |
+
</div>
|
547 |
+
"""
|
548 |
+
history[-1]["content"] = error_message
|
549 |
+
yield history, *[gr.update(visible=False) for _ in range(6)]
|
550 |
+
return
|
551 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
552 |
else:
|
553 |
# Обычный режим работы
|
554 |
if direct_mode:
|
555 |
# Прямая отправка в нейронку для генерации изображения
|
556 |
image_url = generate_image(message)
|
557 |
if not image_url.startswith("Ошибка"):
|
558 |
+
history[-1]["content"] = f""
|
|
|
559 |
else:
|
560 |
history[-1]["content"] = f"❌ {image_url}"
|
561 |
yield history, *[gr.update(visible=False) for _ in range(6)]
|
|
|
577 |
if file_content:
|
578 |
message = f"Файл содержит:\n```\n{file_content}\n```\n\n{message}"
|
579 |
|
580 |
+
# ормируем историю сообщений
|
581 |
for msg in history[:-2]:
|
582 |
messages.append({"role": msg["role"], "content": msg["content"]})
|
583 |
messages.append({"role": "user", "content": str(message)})
|
|
|
602 |
end_idx = partial_message.find("[/GENERATE_IMAGE]")
|
603 |
image_prompt = partial_message[start_idx:end_idx].strip()
|
604 |
|
605 |
+
# Показываем статус генерации
|
606 |
history[-1]["content"] = """
|
607 |
+
<div class="generating-animation">
|
608 |
+
<div class="generating-text">Генерация изображения...</div>
|
609 |
+
</div>
|
610 |
+
"""
|
611 |
yield history, *[gr.update(visible=False) for _ in range(6)]
|
612 |
|
613 |
# Генерируем изображение
|
614 |
image_url = generate_image(image_prompt)
|
615 |
|
616 |
if not image_url.startswith("Ошибка"):
|
617 |
+
# Анализируем изображение
|
618 |
+
# image_analysis = process_image(image_url)
|
619 |
explanation_text = partial_message[end_idx + len("[/GENERATE_IMAGE]"):].strip()
|
620 |
+
partial_message = f"\n\n{explanation_text}"
|
621 |
else:
|
622 |
partial_message = f"❌ {image_url}"
|
623 |
|
|
|
631 |
if code_end != -1:
|
632 |
code_block = partial_message[code_start:code_end].strip()
|
633 |
|
634 |
+
# В конце убирае курсор и сбрасываем файл
|
635 |
history[-1]["content"] = partial_message
|
636 |
+
yield (
|
637 |
+
history,
|
638 |
+
*[gr.update(visible=True if code_block else False) for _ in range(5)],
|
639 |
+
gr.update(value=None) # Сбрасываем файл после ответа
|
640 |
+
)
|
641 |
|
642 |
except Exception as e:
|
643 |
print(f"Error: {e}")
|
|
|
651 |
gr.update(visible=True),
|
652 |
"",
|
653 |
gr.update(visible=False),
|
654 |
+
gr.update(value=None) # Сбрасываем файл в случае ошибки
|
655 |
)
|
656 |
|
657 |
def analyze_code(code):
|
|
|
681 |
return [{"role": "assistant", "content": f"❌ Ошибка при анализе: {str(e)}"}]
|
682 |
|
683 |
def create_interface():
|
684 |
+
# Обновляем тему на темую
|
685 |
theme = gr.themes.Soft(
|
686 |
primary_hue="slate",
|
687 |
secondary_hue="gray",
|
|
|
696 |
input_background_fill="#2d2d2d",
|
697 |
)
|
698 |
|
699 |
+
# Обновляем CSS стили для управления цветом кода
|
700 |
css = """
|
701 |
.gradio-container {
|
702 |
background-color: #1a1a1a !important;
|
|
|
718 |
position: relative;
|
719 |
transition: all 0.3s ease;
|
720 |
}
|
721 |
+
/* Добавляем стиль ля блоков кода */
|
722 |
.message.bot pre {
|
723 |
background-color: #1a1a1a !important;
|
724 |
border-radius: 8px;
|
|
|
950 |
margin-bottom: 10px;
|
951 |
font-style: italic;
|
952 |
}
|
953 |
+
|
954 |
+
.error-animation {
|
955 |
+
position: relative;
|
956 |
+
width: 100%;
|
957 |
+
min-height: 80px;
|
958 |
+
background: linear-gradient(45deg, #2d1a1a, #2d2d2d);
|
959 |
+
border-radius: 15px;
|
960 |
+
overflow: hidden;
|
961 |
+
display: flex;
|
962 |
+
align-items: center;
|
963 |
+
justify-content: center;
|
964 |
+
border-left: 4px solid #e24a4a;
|
965 |
+
padding: 15px;
|
966 |
+
margin: 10px 0;
|
967 |
+
}
|
968 |
+
|
969 |
+
.error-text {
|
970 |
+
color: #ff9999;
|
971 |
+
font-size: 16px;
|
972 |
+
z-index: 2;
|
973 |
+
text-align: center;
|
974 |
+
}
|
975 |
+
|
976 |
+
.thinking-step-block {
|
977 |
+
background: linear-gradient(45deg, #1a1a1a, #2d2d2d);
|
978 |
+
border-left: 4px solid #4a90e2;
|
979 |
+
margin: 15px 0;
|
980 |
+
padding: 15px;
|
981 |
+
border-radius: 8px;
|
982 |
+
animation: fadeIn 0.5s ease-in-out;
|
983 |
+
}
|
984 |
+
|
985 |
+
.final-answer-block {
|
986 |
+
background: linear-gradient(45deg, #1a2a3a, #2d3d4d);
|
987 |
+
border-left: 4px solid #50c878;
|
988 |
+
margin: 20px 0;
|
989 |
+
padding: 20px;
|
990 |
+
border-radius: 8px;
|
991 |
+
font-weight: bold;
|
992 |
+
animation: slideIn 0.5s ease-in-out;
|
993 |
+
}
|
994 |
+
|
995 |
+
@keyframes fadeIn {
|
996 |
+
from {
|
997 |
+
opacity: 0;
|
998 |
+
transform: translateY(10px);
|
999 |
+
}
|
1000 |
+
to {
|
1001 |
+
opacity: 1;
|
1002 |
+
transform: translateY(0);
|
1003 |
+
}
|
1004 |
+
}
|
1005 |
+
|
1006 |
+
@keyframes slideIn {
|
1007 |
+
from {
|
1008 |
+
opacity: 0;
|
1009 |
+
transform: translateX(-10px);
|
1010 |
+
}
|
1011 |
+
to {
|
1012 |
+
opacity: 1;
|
1013 |
+
transform: translateX(0);
|
1014 |
+
}
|
1015 |
+
}
|
1016 |
"""
|
1017 |
|
1018 |
# Добавте этот скрипт в create_interface()
|
|
|
1036 |
render_markdown=True
|
1037 |
)
|
1038 |
|
1039 |
+
# Добавляем компоненты для вывода
|
1040 |
output_text = gr.Textbox(
|
1041 |
label="Вывод",
|
1042 |
interactive=False,
|
|
|
1078 |
|
1079 |
# Добавляем слайдер глубины размышления (изначально скрытый)
|
1080 |
thinking_depth = gr.Slider(
|
1081 |
+
minimum=2.0, # Минимум 3 шага
|
1082 |
+
maximum=25.0,
|
1083 |
+
value=3.0,
|
1084 |
+
step=1.0, # Шаг целочисленный
|
1085 |
+
label="Глубина размышления (количество шагов анализа)",
|
1086 |
visible=False
|
1087 |
)
|
1088 |
|
1089 |
direct_mode = gr.Checkbox(
|
1090 |
+
label="Прямая нерация (без обрботки прмпта)",
|
1091 |
value=False
|
1092 |
)
|
1093 |
file_output = gr.File(label="Загрузить файл")
|
|
|
1199 |
|
1200 |
if __name__ == "__main__":
|
1201 |
demo = create_interface()
|
1202 |
+
|
1203 |
+
|
1204 |
+
# Запускем мониторинг изменений в тдельном потоке
|
1205 |
+
|
1206 |
|
1207 |
+
demo.launch(
|
1208 |
+
show_api=False,
|
1209 |
+
show_error=True,
|
1210 |
+
favicon_path=None,
|
1211 |
+
auth=None,
|
1212 |
+
# quiet=True, # Добавим этот параметр
|
1213 |
+
|
1214 |
+
|
1215 |
+
)
|