Update main.py

main.py

@@ -34,6 +34,11 @@ oauth.register(
     api_base_url='https://www.googleapis.com/oauth2/v1/',
     client_kwargs={'scope': 'openid email profile'}
 )
+@app.get("/login/oauth")
+async def login_oauth(request: Request):
+    # Redirect to OAuth provider (e.g., Google)
+    redirect_uri = request.url_for('auth_callback')
+    return await oauth.google.authorize_redirect(request, redirect_uri)
 
 @app.get("/auth/callback")
 async def auth_callback(request: Request, db: Session = Depends(get_db)):
@@ -43,18 +48,17 @@ async def auth_callback(request: Request, db: Session = Depends(get_db)):
     # Use token to get user info
     user_info = await oauth.google.parse_id_token(request, token)
 
-    # Here, you can check if this user is already in your database,
-    # and if not, create a new user record.
-    # Example:
-    # db_user = db.query(YourUserModel).filter(YourUserModel.email == user_info['email']).first()
-    # if not db_user:
-    #     db_user = YourUserModel(email=user_info['email'], name=user_info['name'])
-    #     db.add(db_user)
-    #     db.commit()
-    
-    # Redirect to a success page, or return a success response
-    return user_info  # or redirect to another page
+    # Check if this user is already in your database, if not, create a new user record
+    db_user = db.query(User).filter(User.email == user_info['email']).first()
+    if not db_user:
+        db_user = User(email=user_info['email'], name=user_info['name'])
+        db.add(db_user)
+        db.commit()
+        db.refresh(db_user)
 
+    # Redirect to a success page, or return a success response
+    # Here you can also create a session or token for your user
+    return {"message": "Login successful", "user": {"email": db_user.email, "name": db_user.name}}
 # Other routes and logic for your application...
 
 app.mount("/static", StaticFiles(directory="static"), name="static")