Update app.py

app.py

@@ -5,7 +5,6 @@ from deep_translator import GoogleTranslator
 from langdetect import detect
 from huggingface_hub import InferenceClient
 import gradio as gr
-import inspect
 
 def get_random_api_key():
     keys = os.getenv("KEYS", "").split(",")
@@ -14,7 +13,6 @@ def get_random_api_key():
     else:
         raise ValueError("API keys not found. Please set the KEYS environment variable.")
 
-
 css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
 try:
     response = requests.get(css_url)
@@ -24,7 +22,6 @@ except requests.exceptions.RequestException as e:
     print(f"Ошибка загрузки CSS: {e}")
     css = " h1{text-align:center} .container { max-width: 800px; margin: 0 auto; }"
 
-
 def generate_story(prompt, style):
     try:
         client = InferenceClient(api_key=get_random_api_key())
@@ -43,8 +40,6 @@ def generate_story(prompt, style):
     except Exception as e:
         return f"Ошибка генерации: {e}"
 
-
-
 def edit_story(original_story, edited_prompt):
     try:
         client = InferenceClient(api_key=get_random_api_key())
@@ -59,8 +54,7 @@ def edit_story(original_story, edited_prompt):
     except Exception as e:
         return f"Ошибка редактирования: {e}"
 
-
-def next_story(original_story, next_prompt):
+def next_story_func(original_story, next_prompt):
     try:
         client = InferenceClient(api_key=get_random_api_key())
         messages = [
@@ -74,8 +68,6 @@ def next_story(original_story, next_prompt):
     except Exception as e:
         return f"Ошибка продления: {e}"
 
-
-
 with gr.Blocks(css=css) as demo:
     with gr.Row():
         style_choices = ["Приключенческая", "Научно-фантастическая", "Романтическая", "Комедийная", "Трагическая", "Введите свой стиль:"]
@@ -94,11 +86,11 @@ 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_output = gr.Textbox(label="Продолжение истории", lines=10)
         next_button = gr.Button("Продлить", variant='primary')
 
-    generate_button.click(generate_story, inputs=[prompt, style], outputs=[output_story], queue=False, api_name="generate_story", concurrency_limit=250)
-    edit_button.click(edit_story, inputs=[output_story, edited_prompt], outputs=[edited_story], queue=False, api_name="edit_story", concurrency_limit=250)
-    next_button.click(next_story, inputs=[output_story, next_prompt], outputs=[next_story], queue=False, api_name="next_story", concurrency_limit=250)
+    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)
 
 demo.launch(show_api=False, share=False)