Update app.py

app.py

@@ -39,7 +39,7 @@ def generate_story(prompt, style):
             {"role": "user", "content": prompt}
         ]
 
-        completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, max_tokens=1000)
+        completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, temperature=0.7, max_tokens=1000)
         story = completion.choices[0].message.content
         return story
     except Exception as e:
@@ -56,7 +56,23 @@ def edit_story(original_story, edited_prompt):
             {"role": "assistant", "content": original_story}
         ]
         
-        completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, max_tokens=1000)
+        completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, temperature=0.7, max_tokens=1000)
+        edited_story = completion.choices[0].message.content
+        return edited_story
+    except Exception as e:
+        return f"Ошибка редактирования: {e}"
+
+def next_story(original_story, edited_prompt=""):
+    try:
+        client = InferenceClient(api_key=get_random_api_key())
+        
+        messages = [
+            {"role": "system", "content": "Продли историю, учитывая предоставленные указания. Продливай В ТОЧНОСТИ С КОНЦА, прям с того же символа, слова, предложения."},
+            {"role": "user", "content": edited_prompt},
+            {"role": "assistant", "content": original_story}
+        ]
+        
+        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
     except Exception as e:
@@ -79,7 +95,13 @@ with gr.Blocks(css=css) as demo:
         edited_story = gr.Textbox(label="Отредактированная история", lines=10)
         edit_button = gr.Button("Отредактировать", variant='primary')
     
+    with gr.Tab("Продление"):
+        next_prompt = gr.Textbox(label="Введите изменения для истории", placeholder="Например: Сделай историю более захватывающей", lines=5)
+        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)
     edit_button.click(edit_story, inputs=[output_story, edited_prompt], outputs=[edited_story], concurrency_limit=250)
+    next_button.click(next_story, inputs=[output_story, next_prompt], outputs=[next_story], concurrency_limit=250)
 
 demo.launch(show_api=False, share=False)