TraceSceneFinal / debug_analysis.py
Siddharth Ravikumar
Add HF token for ZeroGPU auth
287e21d
raw
history blame contribute delete
809 Bytes
import asyncio
from app import run_analysis_fn, create_case_fn, upload_photos_fn, ensure_init
from backend.app.db.database import db
async def setup():
ensure_init()
# Create a case
msg, _ = create_case_fn("TEST-001", "Officer John", "NYC", "2023-01-01", "Notes")
# Add a mock photo
import os
from PIL import Image
os.makedirs("test_photos", exist_ok=True)
img_path = "test_photos/mock.jpg"
Image.new("RGB", (100, 100)).save(img_path)
# upload
upload_photos_fn(1, [img_path])
# Run analysis
try:
res = run_analysis_fn(1)
print("Analysis result:", res)
except Exception as e:
print("Error during analysis:", e)
import traceback
traceback.print_exc()
if __name__ == "__main__":
asyncio.run(setup())