Update main.py

main.py

@@ -13,6 +13,7 @@ from flask_cors import CORS
 import firebase_admin
 from firebase_admin import credentials, db, storage, auth
 from PIL import Image
+from io import BytesIO
 import requests
 
 # Import and configure Google GenAI, matching the Streamlit app