Update app.py

app.py

@@ -14,11 +14,11 @@ except requests.exceptions.RequestException as e:
     css = " h1{text-align:center}"
 
 API_URL = "https://text.pollinations.ai/openai"
-MODEL_NAME = "gemini"
+MODEL_NAME = "gemini"  # Или какая у вас модель
 TEMPERATURE = 0.7
 MAX_TOKENS = 1200
 
-def call_pollinations_api(messages, temperature=TEMPERATURE, max_tokens=MAX_TOKENS, stream=True):
+def call_pollinations_api(messages, temperature=TEMPERATURE, max_tokens=MAX_TOKENS, stream=False): # stream=False
   """Calls the Pollinations API with the given messages."""
   headers = {
       "Content-Type": "application/json"
@@ -30,7 +30,7 @@ def call_pollinations_api(messages, temperature=TEMPERATURE, max_tokens=MAX_TOKE
       "max_tokens": max_tokens,
       "stream": stream
   }
-  response = requests.post(API_URL, headers=headers, data=json.dumps(data), stream=True)
+  response = requests.post(API_URL, headers=headers, data=json.dumps(data))
   response.raise_for_status() # Raise HTTPError for bad responses (4xx or 5xx)
   return response
 
@@ -42,20 +42,10 @@ def generate_story(prompt, style):
         ]
 
         response = call_pollinations_api(messages)
-
-        story = ""
-        for chunk in response.iter_content(chunk_size=512, decode_unicode=True):
-            if chunk: # Filter out keep-alive new chunks
-                if chunk.strip() == "data: [DONE]":
-                    break  # Stop processing when the [DONE] marker is received
-                try:
-                    json_chunk = json.loads(chunk)
-                    if "choices" in json_chunk and len(json_chunk["choices"]) > 0:
-                        story += json_chunk["choices"][0]["delta"].get("content", "") or ""
-                        yield story
-                except json.JSONDecodeError:
-                    print(f"Не удалось декодировать JSON-фрагмент: {chunk}")
-                    yield f"Ошибка декодирования JSON: {chunk}"
+        full_response = response.json()  # Получаем JSON-ответ целиком
+        print("Полный ответ от API:", json.dumps(full_response, indent=2, ensure_ascii=False)) # Выводим отформатированный JSON
+        story = full_response["choices"][0]["message"]["content"] if "choices" in full_response and len(full_response["choices"]) > 0 else "Ответ не получен или структура ответа не соответствует ожидаемой."
+        yield story
 
     except requests.exceptions.RequestException as e:
         yield f"Ошибка API запроса: {e}"
@@ -69,21 +59,11 @@ def edit_story(original_story="", edited_prompt=""):
             {"role": "user", "content": edited_prompt},
             {"role": "user", "content": f"История: ```\n{original_story}\n```"}
         ]
-        response = call_pollinations_api(messages, max_tokens=32760)
-
-        edited_story = ""
-        for chunk in response.iter_content(chunk_size=512, decode_unicode=True):
-            if chunk: # Filter out keep-alive new chunks
-                if chunk.strip() == "data: [DONE]":
-                    break  # Stop processing when the [DONE] marker is received
-                try:
-                    json_chunk = json.loads(chunk)
-                    if "choices" in json_chunk and len(json_chunk["choices"]) > 0:
-                        edited_story += json_chunk["choices"][0]["delta"].get("content", "") or ""
-                        yield edited_story
-                except json.JSONDecodeError:
-                    print(f"Не удалось декодировать JSON-фрагмент: {chunk}")
-                    yield f"Ошибка декодирования JSON: {chunk}"
+        response = call_pollinations_api(messages, max_tokens=32760, stream=False)
+        full_response = response.json()
+        print("Полный ответ от API (edit):", json.dumps(full_response, indent=2, ensure_ascii=False))
+        edited_story = full_response["choices"][0]["message"]["content"] if "choices" in full_response and len(full_response["choices"]) > 0 else "Ответ не получен или структура ответа не соответствует ожидаемой."
+        yield edited_story
 
     except requests.exceptions.RequestException as e:
         yield f"Ошибка API запроса: {e}"
@@ -104,21 +84,11 @@ def next_story_func(original_story="", next_prompt="", continuation_type="Про
             {"role": "user", "content": continuation_prompt},
             {"role": "user", "content": f"История: ```\n{original_story}\n```"}
         ]
-        response = call_pollinations_api(messages)
-
-        next_story = ""
-        for chunk in response.iter_content(chunk_size=512, decode_unicode=True):
-            if chunk: # Filter out keep-alive new chunks
-                if chunk.strip() == "data: [DONE]":
-                    break  # Stop processing when the [DONE] marker is received
-                try:
-                    json_chunk = json.loads(chunk)
-                    if "choices" in json_chunk and len(json_chunk["choices"]) > 0:
-                        next_story += json_chunk["choices"][0]["delta"].get("content", "") or ""
-                        yield next_story
-                except json.JSONDecodeError:
-                    print(f"Не удалось декодировать JSON-фрагмент: {chunk}")
-                    yield f"Ошибка декодирования JSON: {chunk}"
+        response = call_pollinations_api(messages, stream=False)
+        full_response = response.json()
+        print("Полный ответ от API (next):", json.dumps(full_response, indent=2, ensure_ascii=False))
+        next_story = full_response["choices"][0]["message"]["content"] if "choices" in full_response and len(full_response["choices"]) > 0 else "Ответ не получен или структура ответа не соответствует ожидаемой."
+        yield next_story
 
     except requests.exceptions.RequestException as e:
         yield f"Ошибка API запроса: {e}"