AB_AI_RAG_Agent / app /debug.py
kamkol's picture
Added app code to push to HF using Docker
0ee0e61
Raw History Blame Contribute Delete
2.24 kB
import os
import streamlit as st
import pickle
from pathlib import Path
st.title("Debug App")
# Check environment variables
st.write("## Environment Variables")
if os.environ.get("OPENAI_API_KEY"):
st.write("βœ… OPENAI_API_KEY is set")
else:
st.error("❌ OPENAI_API_KEY is not set")
# Try to find data directory
st.write("## Data Directory")
possible_paths = [
"data/processed_data",
"app/data/processed_data",
"/data/processed_data",
"/app/data/processed_data"
]
for path in possible_paths:
if os.path.exists(path):
st.write(f"βœ… Found data directory at: {path}")
# Check for pickle files
chunks_path = os.path.join(path, "chunks.pkl")
if os.path.exists(chunks_path):
st.write(f"βœ… Found chunks.pkl: {os.path.getsize(chunks_path) / (1024*1024):.2f} MB")
# Try to load
try:
with open(chunks_path, "rb") as f:
chunks = pickle.load(f)
st.write(f"βœ… Successfully loaded chunks: {len(chunks)} items")
except Exception as e:
st.error(f"❌ Error loading chunks.pkl: {str(e)}")
else:
st.error(f"❌ chunks.pkl not found in {path}")
embedded_docs_path = os.path.join(path, "embedded_docs.pkl")
if os.path.exists(embedded_docs_path):
st.write(f"βœ… Found embedded_docs.pkl: {os.path.getsize(embedded_docs_path) / (1024*1024):.2f} MB")
# Try to load
try:
with open(embedded_docs_path, "rb") as f:
embedded_docs = pickle.load(f)
st.write(f"βœ… Successfully loaded embedded_docs: {len(embedded_docs)} items")
except Exception as e:
st.error(f"❌ Error loading embedded_docs.pkl: {str(e)}")
else:
st.error(f"❌ embedded_docs.pkl not found in {path}")
break
else:
st.error("❌ Could not find data directory")
st.write("## System Info")
import sys
st.write(f"Python version: {sys.version}")
st.write(f"Working directory: {os.getcwd()}")
st.write(f"Directory contents: {os.listdir('.')}")
st.write("Debug complete.")