Update main.py

main.py

@@ -10,7 +10,7 @@ from flask import Flask, request, jsonify, send_file
 from flask_cors import CORS
 import google.generativeai as genai
 import firebase_admin
-from firebase_admin import credentials, db, storage, auth
+from firebase_admin import credentials, db, storage, auth, firestore
 import pandas as pd
 import numpy as np
 import requests
@@ -35,6 +35,7 @@ except Exception as e:
     print(f"Firebase init error: {e}")
 
 bucket = storage.bucket()
+fs = firestore.client()
 
 # ---------- Helper functions ----------
 def configure_gemini():