Update app.py

app.py

@@ -76,9 +76,10 @@ GENRE_EXAMPLES = {
 # 2. Add constants at the top for magic numbers
 MAX_HISTORY_LENGTH = 20
 MEMORY_WINDOW = 10
-MAX_TOKENS = 1024  # Increased from 512 to allow longer responses
+MAX_TOKENS = 2048  # Doubled for longer responses
 TEMPERATURE = 0.8  # Slightly increased for more creative responses
 TOP_P = 0.95
+MIN_RESPONSE_LENGTH = 200  # Minimum characters before yielding response
 
 def get_examples_for_genre(genre):
     """Get example prompts specific to the selected genre"""
@@ -189,14 +190,11 @@ def respond(
             delta = response_chunk.choices[0].delta.content
             if delta:
                 bot_message += delta
-                # Check response length
-                if len(bot_message.strip()) < 50:  # Minimum response length
-                    continue
-                # Keep storing your conversation as (user, bot) tuples
-                new_history = chat_history.copy()
-                new_history.append((message, bot_message))
-                # Yield in the new dictionary format for Gradio
-                yield format_history_for_gradio(new_history)
+                # Only yield complete responses
+                if len(bot_message.strip()) >= MIN_RESPONSE_LENGTH and "Option 3:" in bot_message:
+                    new_history = chat_history.copy()
+                    new_history.append((message, bot_message))
+                    yield format_history_for_gradio(new_history)
     except Exception as e:
         error_message = f"Story magic temporarily interrupted. Please try again. (Error: {str(e)})"
         broken_history = chat_history + [(message, error_message)]
@@ -222,11 +220,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=3):
             # Chat window + user input
             chatbot = gr.Chatbot(
-                height=500,
-                bubble_full_width=False,
+                height=700,  # Increased height
+                bubble_full_width=True,  # Allow bubbles to use full width
                 show_copy_button=True,
                 avatar_images=(None, "🧙"),
-                type="messages"
+                type="messages",
+                container=True,
+                scale=1,
+                min_width=800  # Ensure minimum width
             )
             msg = gr.Textbox(
                 placeholder="Describe what you want to do next in the story...",