Update app.py

app.py

@@ -1,8 +1,6 @@
 import os
 import random
 import requests
-from deep_translator import GoogleTranslator
-from langdetect import detect
 from huggingface_hub import InferenceClient
 import gradio as gr
 
@@ -26,10 +24,6 @@ def generate_story(prompt, style):
     try:
         client = InferenceClient(api_key=get_random_api_key())
         
-        language = detect(prompt)
-        if language != 'en':
-            prompt = GoogleTranslator(source=language, target='en').translate(prompt)
-
         messages = [
             {"role": "system", "content": f"Напиши историю в стиле {style}."},
             {"role": "user", "content": prompt}
@@ -68,6 +62,12 @@ def next_story_func(original_story, next_prompt):
     except Exception as e:
         return f"Ошибка продления: {e}"
 
+def edone_story(edited_story):
+    return edited_story
+
+def ndone_story(original_story, next_story_output):
+    return original_story + next_story_output
+
 with gr.Blocks(css=css) as demo:
     with gr.Row():
         style_choices = ["Приключенческая", "Научно-фантастическая", "Романтическая", "Комедийная", "Трагическая", "Введите свой стиль:"]
@@ -78,19 +78,27 @@ with gr.Blocks(css=css) as demo:
         generate_button = gr.Button("Сгенерировать историю", variant='primary')
     with gr.Row():
         output_story = gr.Textbox(label="История", lines=10)
+
+    with gr.Accordion("Действия", open=False):
         
-    with gr.Tab("Редактирование"):
-        edited_prompt = gr.Textbox(label="Введите изменения для истории", placeholder="Например: Сделай историю более захватывающей", lines=5)
-        edited_story = gr.Textbox(label="Отредактированная история", lines=10)
-        edit_button = gr.Button("Отредактировать", variant='primary')
+        with gr.Tab("Редактирование"):
+            edited_prompt = gr.Textbox(label="Введите изменения для истории", placeholder="Например: Сделай историю более захватывающей", lines=5)
+            edit_button = gr.Button("Отредактировать", variant='primary')
+            edited_story = gr.Textbox(label="Отредактированная история", lines=10)
+            edone_button = gr.Button("Принять")
 
-    with gr.Tab("Продление"):
-        next_prompt = gr.Textbox(label="Введите изменения для продления истории", placeholder="Например: Продолжи историю", lines=5)
-        next_story_output = gr.Textbox(label="Продолжение истории", lines=10)
-        next_button = gr.Button("Продлить", variant='primary')
+        with gr.Tab("Продление"):
+            next_prompt = gr.Textbox(label="Введите изменения для продления истории (Необязательно)", placeholder="Продолжи, но чтобы было...", lines=5)
+            next_button = gr.Button("Продлить", variant='primary')
+            next_story_output = gr.Textbox(label="Продолжение истории", lines=10)
+            ndone_button = gr.Button("Принять")
 
     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_func, inputs=[output_story, next_prompt], outputs=[next_story_output], concurrency_limit=250)
+    
+    edone_button.click(edone_story, inputs=[edited_story], outputs=[output_story], concurrency_limit=550)
+    
+    ndone_button.click(ndone_story, inputs=[output_story, next_story_output], outputs=[output_story], concurrency_limit=550)
 
 demo.launch(show_api=False, share=False)