rairo commited on
Commit
e15a876
·
verified ·
1 Parent(s): 53cc600

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +15 -6
main.py CHANGED
@@ -107,22 +107,31 @@ def signup():
107
  def get_user_profile():
108
  try:
109
  auth_header = request.headers.get('Authorization', '')
 
 
110
  if not auth_header.startswith('Bearer '):
111
- return jsonify({'error': 'Authorization header missing or malformed'}), 401
 
112
  token = auth_header.split(' ')[1]
113
  uid = verify_token(token)
 
114
  if not uid:
115
- return jsonify({'error': 'Invalid token'}), 401
116
- user_record = auth.get_user(uid)
117
  user_data = db.reference(f'users/{uid}').get()
 
 
 
 
 
118
  return jsonify({
119
  'uid': uid,
120
- 'email': user_record.email,
121
  'credits': user_data.get('credits', 0),
122
- 'is_admin': user_data.get('is_admin', False),
123
- 'created_at': user_data.get('created_at')
124
  })
125
  except Exception as e:
 
126
  return jsonify({'error': str(e)}), 500
127
 
128
  # ---------- Video Generation Endpoint ----------
 
107
  def get_user_profile():
108
  try:
109
  auth_header = request.headers.get('Authorization', '')
110
+ print("Received Auth Header:", auth_header) # Debugging
111
+
112
  if not auth_header.startswith('Bearer '):
113
+ return jsonify({'error': 'Missing or invalid token'}), 401
114
+
115
  token = auth_header.split(' ')[1]
116
  uid = verify_token(token)
117
+
118
  if not uid:
119
+ return jsonify({'error': 'Invalid or expired token'}), 401
120
+
121
  user_data = db.reference(f'users/{uid}').get()
122
+ print("Fetched User Data:", user_data) # Debugging
123
+
124
+ if not user_data:
125
+ return jsonify({'error': 'User not found'}), 404
126
+
127
  return jsonify({
128
  'uid': uid,
129
+ 'email': user_data.get('email'),
130
  'credits': user_data.get('credits', 0),
131
+ 'is_admin': user_data.get('is_admin', False)
 
132
  })
133
  except Exception as e:
134
+ print(f"Error fetching user profile: {str(e)}")
135
  return jsonify({'error': str(e)}), 500
136
 
137
  # ---------- Video Generation Endpoint ----------