Update app.py

app.py

@@ -5,24 +5,23 @@ 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(",")
-    if keys and keys[0]:  # Check if KEYS is set and not empty
+    if keys and keys[0]:
         return random.choice(keys).strip()
     else:
         raise ValueError("API keys not found. Please set the KEYS environment variable.")
 
-# Ссылка на файл CSS
-css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
 
-# Получение CSS по ссылке
+css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
 try:
     response = requests.get(css_url)
     response.raise_for_status()
     css = response.text + " h1{text-align:center} .container { max-width: 800px; margin: 0 auto; }"
 except requests.exceptions.RequestException as e:
-    print(f"Ошибка при загрузке CSS: {e}")
+    print(f"Ошибка загрузки CSS: {e}")
     css = " h1{text-align:center} .container { max-width: 800px; margin: 0 auto; }"
 
 
@@ -33,12 +32,11 @@ def generate_story(prompt, style):
         language = detect(prompt)
         if language != 'en':
             prompt = GoogleTranslator(source=language, target='en').translate(prompt)
-        
+
         messages = [
             {"role": "system", "content": f"Напиши историю в стиле {style}."},
             {"role": "user", "content": prompt}
         ]
-
         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
@@ -46,38 +44,38 @@ def generate_story(prompt, style):
         return f"Ошибка генерации: {e}"
 
 
+
 def edit_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=1000)
         edited_story = completion.choices[0].message.content
         return edited_story
     except Exception as e:
         return f"Ошибка редактирования: {e}"
 
-def next_story(original_story, next_prompt=""):
+
+def next_story(original_story, next_prompt):
     try:
         client = InferenceClient(api_key=get_random_api_key())
-        
         messages = [
             {"role": "system", "content": "Продли историю, учитывая предоставленные указания. Продливай В ТОЧНОСТИ С КОНЦА, прям с того же символа, слова, предложения."},
-            {"role": "user", "content": edited_prompt},
+            {"role": "user", "content": next_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)
+        completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, temperature=0.7, max_tokens=1000)
         next_story = completion.choices[0].message.content
         return next_story
     except Exception as e:
         return f"Ошибка продления: {e}"
 
+
+
 with gr.Blocks(css=css) as demo:
     with gr.Row():
         style_choices = ["Приключенческая", "Научно-фантастическая", "Романтическая", "Комедийная", "Трагическая", "Введите свой стиль:"]
@@ -86,22 +84,21 @@ with gr.Blocks(css=css) as demo:
         prompt = gr.Textbox(label="Введите запрос для истории", placeholder="Например: История о путешествии в космос", lines=5)
     with gr.Row():
         generate_button = gr.Button("Сгенерировать историю", variant='primary')
-        
     with gr.Row():
         output_story = gr.Textbox(label="История", lines=10)
-    
+        
     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("Продление"):
-        next_prompt = gr.Textbox(label="Введите изменения для истории", placeholder="Например: Сделай историю более захватывающей", lines=5)
+        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)
+    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)
+
+demo.launch(show_api=False, share=False)