Update app.py

app.py

@@ -16,7 +16,7 @@ except requests.exceptions.RequestException as e:
 API_URL = "https://text.pollinations.ai/openai"
 MODEL_NAME = "gemini"
 TEMPERATURE = 0.7
-MAX_TOKENS = 1500
+MAX_TOKENS = 1200
 
 def call_pollinations_api(messages, temperature=TEMPERATURE, max_tokens=MAX_TOKENS, stream=True):
   """Calls the Pollinations API with the given messages."""
@@ -46,6 +46,8 @@ def generate_story(prompt, style):
         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:
@@ -61,13 +63,6 @@ def generate_story(prompt, style):
         yield f"Ошибка генерации: {e}"
 
 def edit_story(original_story="", edited_prompt=""):
-    if original_story == "":
-        yield f"Сначала сгенерируйте историю!"
-        return
-    if edited_prompt == "":
-        yield f"Для начала введите что изменить."
-        return
-        
     try:
         messages = [
             {"role": "system", "content": "Отредактируй историю, учитывая предоставленные указания. Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."},
@@ -79,6 +74,8 @@ def edit_story(original_story="", edited_prompt=""):
         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:
@@ -94,10 +91,6 @@ def edit_story(original_story="", edited_prompt=""):
         yield f"Ошибка редактирования: {e}"
 
 def next_story_func(original_story="", next_prompt="", continuation_type="Продолжение"):
-    if original_story == "":
-        yield f"Сначала сгенерируйте историю!"
-        return
-    
     try:
         if continuation_type == "Продолжение следует...":
             continuation_prompt = "Продолжи, подведя к тому, что продолжение следует."
@@ -116,6 +109,8 @@ def next_story_func(original_story="", next_prompt="", continuation_type="Про
         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: