Update app.py

app.py

@@ -70,7 +70,10 @@ def next_story_func(original_story="", next_prompt=""):
     except Exception as e:
         return f"Ошибка продления: {e}"
 
-def edone_story(edited_story):
+def edone_story(original_story="", edited_story=""):
+    if edited_story == "":
+        return original_story
+    
     return edited_story
 
 def ndone_story(original_story, next_story_output):
@@ -108,7 +111,7 @@ with gr.Blocks(css=css) as demo:
     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)
+    edone_button.click(edone_story, inputs=[output_story, 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)