fix_titles.py

from backend.database import SessionLocal, text
from sqlalchemy.orm import sessionmaker

def fix_broken_titles():
    db = SessionLocal()
    try:
        # Find titles starting with "Error calling LLM" or "Request timed out"
        query = text("SELECT id, title FROM chat_sessions WHERE title LIKE 'Error calling LLM%' OR title LIKE '%Request timed out%'")
        params = {}
        result = db.execute(query, params).fetchall()
        
        print(f"Found {len(result)} broken titles.")
        for row in result:
            print(f"Fixing Session {row[0]}: {row[1]}")
            update_query = text("UPDATE chat_sessions SET title = 'New Chat' WHERE id = :id")
            db.execute(update_query, {'id': row[0]})
        
        db.commit()
        print("Done.")
    except Exception as e:
        print(f"Error: {e}")
        db.rollback()
    finally:
        db.close()

if __name__ == "__main__":
    fix_broken_titles()