Update app.py

app.py

@@ -5,8 +5,11 @@ import random
 from typing import Generator, Dict, List, Tuple, Optional
 import logging  # Added logging for better debugging
 
-# Configure logging
-logging.basicConfig(level=logging.ERROR, format='%(asctime)s - %(levelname)s - %(message)s')
+# Configure logging with DEBUG level
+logging.basicConfig(
+    level=logging.DEBUG,
+    format='%(asctime)s - %(levelname)s - %(message)s'
+)
 
 # Get token from environment variable
 hf_token = os.environ.get("HF_TOKEN")
@@ -135,48 +138,50 @@ def format_history_for_gradio(history_tuples):
 
 # 1. Add type hints for better code maintainability
 # 4. Add input validation
-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]]]:
+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]]]:
     """Generate a response based on the current message and conversation history."""
     if not message.strip():
         return "", chat_history
     
     try:
-        # Start with the system prompt
+        # Start with system prompt
         api_messages = [{"role": "system", "content": get_enhanced_system_prompt(genre)}]
+        logging.debug(f"System Message: {api_messages[0]}")
         
-        # Add formatted chat history
+        # Add chat history
         if chat_history and use_full_memory:
             for user_msg, bot_msg in chat_history[-MEMORY_WINDOW:]:
-                api_messages.append({"role": "user", "content": str(user_msg)})
-                api_messages.append({"role": "assistant", "content": str(bot_msg)})
+                api_messages.extend([
+                    {"role": "user", "content": str(user_msg)},
+                    {"role": "assistant", "content": str(bot_msg)}
+                ])
+            logging.debug(f"Chat History Messages: {api_messages[1:]}")
         
-        # Add the user's latest message
+        # Add current message
         api_messages.append({"role": "user", "content": str(message)})
+        logging.debug(f"Final Message List: {api_messages}")
 
-        # Make the API call
+        # Make API call
+        logging.debug("Making API call...")
         response = client.chat_completion(
             messages=api_messages,
             max_tokens=MAX_TOKENS,
             temperature=TEMPERATURE,
             top_p=TOP_P
         )
+        logging.debug("API call completed")
         
-        # Extract response content
+        # Extract response
         bot_message = response.choices[0].message.content
+        logging.debug(f"Bot Response: {bot_message[:100]}...")  # First 100 chars
         
-        # Update chat history
+        # Update history
         updated_history = chat_history + [(message, bot_message)]
         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)
+        error_msg = f"Story magic temporarily interrupted. Please try again. (Error: {str(e)})"
         return "", chat_history + [(message, error_msg)]
 
 def save_story(chat_history):