Update main.py

main.py

@@ -162,26 +162,58 @@ def ensure_user_profile_exists(uid, email=None):
                 'suspended': False
             }
             
-            # Use transaction to ensure atomic write
-            def create_profile_transaction(current_data):
-                if current_data is None:
-                    return new_profile
-                return current_data  # Profile already exists
-            
-            result = user_ref.transaction(create_profile_transaction)
-            if result.committed:
-                logging.info(f"Successfully created profile for UID: {uid}")
-                return new_profile
-            else:
-                logging.error(f"Failed to create profile for UID: {uid} - transaction not committed")
-                return None
+            try:
+                # Use transaction to ensure atomic write
+                def create_profile_transaction(current_data):
+                    if current_data is None:
+                        return new_profile
+                    return current_data  # Profile already exists
+                
+                # Transaction returns the final data, not a result object
+                final_data = user_ref.transaction(create_profile_transaction)
+                
+                # Verify the profile was created by checking if it matches our new profile
+                if final_data and final_data.get('email') == email:
+                    logging.info(f"Successfully created profile for UID: {uid}")
+                    return final_data
+                else:
+                    logging.error(f"Transaction completed but profile verification failed for UID: {uid}")
+                    # Fall back to direct set if transaction didn't work as expected
+                    user_ref.set(new_profile)
+                    
+                    # Verify the direct set worked
+                    verification = user_ref.get()
+                    if verification:
+                        logging.info(f"Profile created via fallback method for UID: {uid}")
+                        return verification
+                    else:
+                        logging.error(f"Both transaction and direct set failed for UID: {uid}")
+                        return None
+                        
+            except Exception as e:
+                logging.error(f"Transaction failed for UID {uid}, attempting direct set: {e}")
+                try:
+                    # Fallback to direct set
+                    user_ref.set(new_profile)
+                    verification = user_ref.get()
+                    if verification:
+                        logging.info(f"Profile created via fallback for UID: {uid}")
+                        return verification
+                    else:
+                        logging.error(f"Fallback set also failed for UID: {uid}")
+                        return None
+                except Exception as e2:
+                    logging.error(f"Both transaction and fallback failed for UID {uid}: {e2}")
+                    return None
         
         return user_data
         
     except Exception as e:
         logging.error(f"Error ensuring profile exists for UID {uid}: {e}")
+        logging.error(traceback.format_exc())
         return None
 
+
 # Updated signup function with better error handling
 @app.route('/api/auth/signup', methods=['POST'])
 def signup():