Update main.py

main.py

@@ -11,6 +11,7 @@ import google.generativeai as genai
 import firebase_admin
 from firebase_admin import credentials, db, storage, auth
 import pandas as pd
+import requests  # Needed for calling Firebase REST API
 
 app = Flask(__name__)
 CORS(app)
@@ -24,6 +25,7 @@ firebase_admin.initialize_app(cred, {
 
 bucket = storage.bucket()
 api_key = os.environ['Gemini']
+FIREBASE_API_KEY = os.environ.get('FIREBASE_API_KEY')  # For auth REST API calls
 
 # Helper functions
 def configure_gemini():
@@ -54,7 +56,15 @@ def check_daily_reset(user_ref):
         print(f"Reset error: {str(e)}")
         return False
 
-# Enhanced receipt processing endpoint
+# ========================================
+# Authentication Endpoints
+# ========================================
+
+
+# ========================================
+# Receipt Processing Endpoint (unchanged)
+# ========================================
+
 @app.route('/api/process-receipt', methods=['POST'])
 def process_receipt_endpoint():
     try:
@@ -175,7 +185,10 @@ def validate_and_save_transaction(uid, user_data, data, file_hash, image_bytes,
         'remaining_cash': new_remaining
     })
 
-# Enhanced data endpoints for visualizations
+# ========================================
+# Data Endpoints for Visualizations
+# ========================================
+
 @app.route('/api/user/spending-overview', methods=['GET'])
 def get_spending_overview():
     try:
@@ -232,7 +245,7 @@ def verify_admin(auth_header):
     if not user.custom_claims or not user.custom_claims.get('admin'):
         raise PermissionError('Admin access required')
 
-# User management endpoints
+# User management endpoint for profile
 @app.route('/api/user/profile', methods=['GET'])
 def get_user_profile():
     try:
@@ -252,4 +265,4 @@ def get_user_profile():
         return jsonify({'error': str(e)}), 500
 
 if __name__ == '__main__':
-    app.run(debug=True, host="0.0.0.0", port=7860)
+    app.run(debug=True, host="0.0.0.0", port=7860)