Update app.py

app.py

@@ -4,12 +4,15 @@ from duckduckgo_search import DDGS
 
 # ==== Константы ====
 MODEL_NAME = "openai/gpt-oss-20b"
-SYSTEM_MESSAGE = """SYSTEM_MESSAGE = Ты Pok.Bot, ты используешь открытую локальную модель GPT-OSS от OpenAI, не GPT-4 (наверно😆). Тебя создал POKilondron. Используй емодзи 😄.If you don't know the answer, output a command like this:#search <query> Do NOT invent facts. И да использую команду #search «запрос» когда нужно и не повторяй постоянноодин и тоже запрос и не видавай запрос на 18+ 😠. Тебя обновили в 15 сентября в 2025 году, теперьты можешь взаимодействовать с поиском duckduckgo через команду"""
-MAX_TOKENS = 100000
+IMAGE_MODEL = "stabilityai/stable-diffusion-2-1"
+SYSTEM_MESSAGE = """SYSTEM_MESSAGE = Ты Pok.Bot, ты используешь открытую локальную модель GPT-OSS от OpenAI, не GPT-4 (наверно😆). Тебя создал POKilondron. Используй емодзи 😄.И да использую команду #search «запрос» когда нужно и не повторяй постоянноодин и тоже запрос и не видавай запрос на 18+ 😠. Тебя обновили в 20 сентября в 2025 году, теперьты можешь генерировать картинки и еще с обновлениям 15 сентебря ти можешь взаимодействовать с поиском duckduckgo через команду #search
+If you don't know the answer, output a command like this: #search <query>.
+If the user asks to generate an image, output a command like this: #image <description>.
+Do NOT invent facts. Always use the given Web context if available."""
+MAX_TOKENS = 512
 TEMPERATURE = 0.7
 TOP_P = 0.95
 
-
 # ==== Поиск через DuckDuckGo ====
 def search_web(query: str, max_results: int = 3):
     results = []
@@ -18,7 +21,6 @@ def search_web(query: str, max_results: int = 3):
             results.append(f"{r['title']}: {r['body']}")
     return "\n".join(results) if results else "Ничего не найдено."
 
-
 # ==== Основная функция ====
 def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
     client = InferenceClient(token=hf_token.token, model=MODEL_NAME)
@@ -30,11 +32,8 @@ def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
 
     raw_response = ""
     for msg in client.chat_completion(
-        messages,
-        max_tokens=MAX_TOKENS,
-        stream=True,
-        temperature=TEMPERATURE,
-        top_p=TOP_P,
+        messages, max_tokens=MAX_TOKENS, stream=True,
+        temperature=TEMPERATURE, top_p=TOP_P,
     ):
         if not msg.choices:
             continue
@@ -47,7 +46,6 @@ def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
         query = raw_response.replace("#search", "").strip()
         web_context = search_web(query)
 
-        # 2-й прогон: модель формирует финальный ответ с учётом поиска
         followup_messages = [
             {"role": "system", "content": SYSTEM_MESSAGE},
             {"role": "system", "content": f"Web context:\n{web_context}"},
@@ -57,20 +55,27 @@ def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
 
         final_response = ""
         for msg in client.chat_completion(
-            followup_messages,
-            max_tokens=MAX_TOKENS,
-            stream=True,
-            temperature=TEMPERATURE,
-            top_p=TOP_P,
+            followup_messages, max_tokens=MAX_TOKENS, stream=True,
+            temperature=TEMPERATURE, top_p=TOP_P,
         ):
             if not msg.choices:
                 continue
             delta = msg.choices[0].delta.content
             if delta:
                 final_response += delta
-                yield final_response
+        yield final_response
+
+    # === Проверка на команду картинки ===
+    elif raw_response.strip().startswith("#image"):
+        prompt = raw_response.replace("#image", "").strip()
+        img_client = InferenceClient(token=hf_token.token, model=IMAGE_MODEL)
+
+        image = img_client.text_to_image(prompt, size="512x512")
+        image.save("generated.png")
+        yield (f"Вот сгенерированное изображение по запросу: '{prompt}'", "generated.png")
+
     else:
-        # Если поиска не нужно — выдаём обычный ответ
+        # Если поиска/картинки не нужно — обычный ответ
         yield raw_response