Update app.py

app.py

@@ -103,10 +103,28 @@ def create_story_summary(chat_history):
     }
     return summary_instruction
 
+# NEW HELPER: Convert (user, bot) tuples into the "role/content" dictionaries Gradio needs
+def format_history_for_gradio(history_tuples):
+    """
+    Convert a list of (user_msg, bot_msg) tuples
+    into a list of dicts for 'type="messages"':
+    [
+      {"role": "user", "content": "foo"},
+      {"role": "assistant", "content": "bar"},
+      ...
+    ]
+    """
+    messages = []
+    for user_msg, bot_msg in history_tuples:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+    return messages
+
 def respond(message, chat_history, genre=None, use_full_memory=True):
     """Generate a response based on the current message and conversation history"""
     system_message = get_enhanced_system_prompt(genre)
 
+    # Convert your existing (user, bot) history into a format for the API request
     formatted_history = []
     for user_msg, bot_msg in chat_history:
         formatted_history.append({"role": "user", "content": user_msg})
@@ -114,8 +132,9 @@ def respond(message, chat_history, genre=None, use_full_memory=True):
     
     api_messages = [{"role": "system", "content": system_message}]
     
+    # Use full memory or partial memory
     if use_full_memory and formatted_history:
-        if len(formatted_history) > 20:  # Arbitrary threshold
+        if len(formatted_history) > 20:
             summary_instruction = create_story_summary(chat_history[:len(chat_history)-5])
             if summary_instruction:
                 api_messages.append(summary_instruction)
@@ -130,8 +149,10 @@ def respond(message, chat_history, genre=None, use_full_memory=True):
             for msg in formatted_history[-memory_length*2:]:
                 api_messages.append(msg)
     
+    # Add current user message
     api_messages.append({"role": "user", "content": message})
     
+    # Special handling for story initialization
     if not chat_history or message.lower() in ["start", "begin", "begin my adventure"]:
         api_messages.append({
             "role": "system", 
@@ -150,12 +171,15 @@ def respond(message, chat_history, genre=None, use_full_memory=True):
             delta = response_chunk.choices[0].delta.content
             if delta:
                 bot_message += delta
+                # Keep storing your conversation as (user, bot) tuples
                 new_history = chat_history.copy()
                 new_history.append((message, bot_message))
-                yield new_history
+                # Yield in the new dictionary format for Gradio
+                yield format_history_for_gradio(new_history)
     except Exception as e:
         error_message = f"Story magic temporarily interrupted. Please try again. (Error: {str(e)})"
-        yield chat_history + [(message, error_message)]
+        broken_history = chat_history + [(message, error_message)]
+        yield format_history_for_gradio(broken_history)
 
 def save_story(chat_history):
     """Convert chat history to markdown for download"""
@@ -222,7 +246,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             def update_starter_buttons(selected_genre):
                 # Grab up to 4 examples from the chosen genre
                 examples = get_examples_for_genre(selected_genre)
-                # If there are fewer than 4, fill the rest with placeholders or hide them
                 button_updates = []
                 for i in range(4):
                     if i < len(examples):
@@ -233,9 +256,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             
             # Function that populates the msg with the chosen starter text, then calls respond
             def pick_starter(starter_text, chat_history, selected_genre, memory_flag):
-                # The function returns the text to put in msg
-                # but we also chain the respond function via .then
-                return starter_text
+                return starter_text  # This goes into 'msg'
             
             # Hook each starter button:
             #  1) Put the chosen text into 'msg'
@@ -273,7 +294,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         save_btn = gr.Button("Download My Story", variant="secondary")
         story_output = gr.Markdown(visible=False)
     
-    # "Download My Story" logic
     save_btn.click(save_story, inputs=[chatbot], outputs=[story_output])
     save_btn.click(
         fn=lambda: True,