Update app.py

app.py

@@ -138,7 +138,7 @@ def format_history_for_gradio(history_tuples):
 def respond(
     message: str, 
     chat_history: List[Tuple[str, str]], 
-    genre: Optional[str] = None,
+    genre: Optional[str] = None, 
     use_full_memory: bool = True
 ) -> Tuple[str, List[Tuple[str, str]]]:
     """Generate a response based on the current message and conversation history."""
@@ -146,21 +146,19 @@ def respond(
         return "", chat_history
     
     try:
-        # Format messages for API
+        # Start with the system prompt
         api_messages = [{"role": "system", "content": get_enhanced_system_prompt(genre)}]
         
-        # Add chat history
+        # Add formatted chat history
         if chat_history and use_full_memory:
             for user_msg, bot_msg in chat_history[-MEMORY_WINDOW:]:
-                api_messages.extend([
-                    {"role": "user", "content": str(user_msg)},
-                    {"role": "assistant", "content": str(bot_msg)}
-                ])
+                api_messages.append({"role": "user", "content": str(user_msg)})
+                api_messages.append({"role": "assistant", "content": str(bot_msg)})
         
-        # Add current message
+        # Add the user's latest message
         api_messages.append({"role": "user", "content": str(message)})
 
-        # Make API call
+        # Make the API call
         response = client.chat_completion(
             messages=api_messages,
             max_tokens=MAX_TOKENS,
@@ -168,18 +166,17 @@ def respond(
             top_p=TOP_P
         )
         
-        # Get response content
+        # Extract response content
         bot_message = response.choices[0].message.content
         
-        # Update chat history with new exchange
+        # Update chat history
         updated_history = chat_history + [(message, bot_message)]
-        
-        # Return updated components
         return "", updated_history
-                        
+    
     except Exception as e:
+        # Enhanced error handling
         error_msg = f"Story magic temporarily interrupted. Please try again. (Error: {str(e)})"
-        logging.error("Error in respond function", exc_info=True)  # Added logging
+        logging.error("Error in respond function", exc_info=True)
         return "", chat_history + [(message, error_msg)]
 
 def save_story(chat_history):