Update app.py

app.py

@@ -125,8 +125,9 @@ def create_story_summary(chat_history):
         return None
     
     story_text = ""
-    for user_msg, bot_msg in chat_history:
-        story_text += f"User: {user_msg}\nStory: {bot_msg}\n\n"
+    for i in range(0, len(chat_history), 2):
+        if i+1 < len(chat_history):
+            story_text += f"User: {chat_history[i]}\nStory: {chat_history[i+1]}\n\n"
     
     summary_instruction = {
         "role": "system",
@@ -134,8 +135,8 @@ def create_story_summary(chat_history):
     }
     return summary_instruction
 
-# Modified function for proper Gradio format (tuples)
-def respond(message: str, chat_history: List[Tuple[str, str]], genre: Optional[str] = None, use_full_memory: bool = True) -> Tuple[str, List[Tuple[str, str]]]:
+# Modified function for proper Gradio format (lists)
+def respond(message: str, chat_history: List[str], genre: Optional[str] = None, use_full_memory: bool = True) -> Tuple[str, List[str]]:
     """Generate a response based on the current message and conversation history."""
     if not message.strip():
         return "", chat_history
@@ -145,9 +146,11 @@ def respond(message: str, chat_history: List[Tuple[str, str]], genre: Optional[s
         api_messages = [{"role": "system", "content": get_enhanced_system_prompt(genre)}]
         logging.debug(f"System Message: {api_messages[0]}")
         
-        # Add chat history - convert from tuples to API format
+        # Add chat history - convert from lists to API format
         if chat_history and use_full_memory:
-            for user_msg, bot_msg in chat_history[-MEMORY_WINDOW:]:
+            # Take pairs of messages (user, bot) from history
+            history_pairs = list(zip(chat_history[::2], chat_history[1::2]))
+            for user_msg, bot_msg in history_pairs[-MEMORY_WINDOW:]:
                 api_messages.extend([
                     {"role": "user", "content": str(user_msg)},
                     {"role": "assistant", "content": str(bot_msg)}
@@ -172,14 +175,14 @@ def respond(message: str, chat_history: List[Tuple[str, str]], genre: Optional[s
         bot_message = response.choices[0].message.content
         logging.debug(f"Bot Response: {bot_message[:100]}...")  # First 100 chars
         
-        # Update history using tuple format (user_msg, bot_msg)
-        updated_history = chat_history + [(message, bot_message)]
+        # Update history using list format [user_msg, bot_msg, user_msg, bot_msg, ...]
+        updated_history = chat_history + [message, bot_message]
         return "", updated_history
     
     except Exception as e:
         logging.error("Error in respond function", exc_info=True)
         error_msg = f"Story magic temporarily interrupted. Please try again. (Error: {str(e)})"
-        return "", chat_history + [(message, error_msg)]
+        return "", chat_history + [message, error_msg]
 
 def save_story(chat_history):
     """Convert chat history to markdown and return as downloadable file"""
@@ -187,9 +190,11 @@ def save_story(chat_history):
         return None
     
     story_text = "# My Interactive Adventure\n\n"
-    for user_msg, bot_msg in chat_history:
-        story_text += f"**Player:** {user_msg}\n\n"
-        story_text += f"**Story:** {bot_msg}\n\n---\n\n"
+    # Process in pairs (user message, bot response)
+    for i in range(0, len(chat_history), 2):
+        if i+1 < len(chat_history):
+            story_text += f"**Player:** {chat_history[i]}\n\n"
+            story_text += f"**Story:** {chat_history[i+1]}\n\n---\n\n"
     
     # Return as a file for download
     return story_text
@@ -216,17 +221,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     > **Tip:** The more detail you provide, the deeper the story becomes.
     """)
     
-    # Remove the old status message and description
-    # with gr.Row():
-    #     status_message = gr.Markdown("Ready to begin your adventure...", visible=True)
-    # gr.Markdown("Create a completely unique literary world...")
-    
-    # Get avatar URL and continue with rest of interface
     wizard_avatar = get_storyteller_avatar_url()
     
     with gr.Row():
         with gr.Column(scale=3):
-            # Chat window + user input - USING TUPLES FORMAT
+            # Chat window + user input - USING LIST FORMAT
             chatbot = gr.Chatbot(
                 height=500,
                 bubble_full_width=True,
@@ -235,7 +234,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 container=True,
                 scale=1,
                 min_width=800,
-                value=[],  # Empty list of tuples
+                value=[],  # Empty list for messages
                 render=True
             )
             msg = gr.Textbox(
@@ -300,7 +299,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                                 
                 except Exception as e:
                     error_msg = f"Story magic temporarily interrupted. Please try again. (Error: {str(e)})"
-                    return "", history + [(starter_text, error_msg)]
+                    return "", history + [starter_text, error_msg]
 
             # Simplified button connections
             for starter_button in starter_buttons: