jdesiree commited on
Commit
13b2a24
·
verified ·
1 Parent(s): 04b9ca0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py CHANGED
@@ -60,6 +60,34 @@ load_dotenv(".env")
60
  HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACEHUB_API_TOKEN")
61
  print("Environment variables loaded.")
62
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
63
  # Setup main logger first
64
  logging.basicConfig(level=logging.INFO)
65
  logger = logging.getLogger(__name__)
 
60
  HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACEHUB_API_TOKEN")
61
  print("Environment variables loaded.")
62
 
63
+ DEBUG_STATE = os.getenv("DEBUG_STATE", "false").lower() == "true"
64
+
65
+ def debug_state(conversation_state, event_name="", force_debug=False):
66
+ """Debug function to inspect current conversation state"""
67
+ if not (DEBUG_STATE or force_debug):
68
+ return conversation_state
69
+
70
+ timestamp = datetime.now().strftime("%H:%M:%S")
71
+ logger.info(f"[{timestamp}] DEBUG STATE - {event_name}")
72
+ logger.info(f"Total messages: {len(conversation_state)}")
73
+
74
+ for i, msg in enumerate(conversation_state):
75
+ role = msg["role"]
76
+ content_preview = msg["content"][:100] + "..." if len(msg["content"]) > 100 else msg["content"]
77
+ logger.info(f" {i+1}. {role}: {content_preview}")
78
+
79
+ # Log to file for later analysis
80
+ if DEBUG_STATE:
81
+ debug_log_file = "debug_state.log"
82
+ with open(debug_log_file, "a", encoding="utf-8") as f:
83
+ f.write(f"\n=== {timestamp} - {event_name} ===\n")
84
+ f.write(f"Total messages: {len(conversation_state)}\n")
85
+ for i, msg in enumerate(conversation_state):
86
+ f.write(f"{i+1}. {msg['role']}: {msg['content'][:200]}...\n")
87
+ f.write("=" * 40 + "\n")
88
+
89
+ return conversation_state
90
+
91
  # Setup main logger first
92
  logging.basicConfig(level=logging.INFO)
93
  logger = logging.getLogger(__name__)