Update app.py

app.py

@@ -34,7 +34,12 @@ def generate_story(prompt, style):
     except Exception as e:
         return f"Ошибка генерации: {e}"
 
-def edit_story(original_story, edited_prompt):
+def edit_story(original_story="", edited_prompt=""):
+    if original_story == "":
+        return f"Сначала сгенерируйте историю!"
+    if edited_prompt == "":
+        return f"Для начала введите что изменить."
+        
     try:
         client = InferenceClient(api_key=get_random_api_key())
         messages = [
@@ -48,7 +53,10 @@ def edit_story(original_story, edited_prompt):
     except Exception as e:
         return f"Ошибка редактирования: {e}"
 
-def next_story_func(original_story, next_prompt):
+def next_story_func(original_story="", next_prompt):
+    if original_story == "":
+        return f"Сначала сгенерируйте историю!"
+    
     try:
         client = InferenceClient(api_key=get_random_api_key())
         messages = [