Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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__)
|