Update app.py

app.py

@@ -62,7 +62,7 @@ def edit_story(original_story, edited_prompt):
     except Exception as e:
         return f"Ошибка редактирования: {e}"
 
-def next_story(original_story, edited_prompt=""):
+def next_story(original_story, next_prompt=""):
     try:
         client = InferenceClient(api_key=get_random_api_key())
         
@@ -73,10 +73,10 @@ def next_story(original_story, edited_prompt=""):
         ]
         
         completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, temperature=0.7, max_tokens=2500)
-        edited_story = completion.choices[0].message.content
-        return edited_story
+        next_story = completion.choices[0].message.content
+        return next_story
     except Exception as e:
-        return f"Ошибка редактирования: {e}"
+        return f"Ошибка продления: {e}"
 
 with gr.Blocks(css=css) as demo:
     with gr.Row():
@@ -97,7 +97,7 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Tab("Продление"):
         next_prompt = gr.Textbox(label="Введите изменения для истории", placeholder="Например: Сделай историю более захватывающей", lines=5)
-        next_story = gr.Textbox(label="Отредактированная история", lines=10)
+        next_story = gr.Textbox(label="Продолжение истории", lines=10)
         next_button = gr.Button("Продлить", variant='primary')
     
     generate_button.click(generate_story, inputs=[prompt, style], outputs=[output_story], concurrency_limit=250)