Update main.py

main.py

@@ -94,9 +94,26 @@ async def register_post(
     confirm_password: str = Form(...),
     db: Session = Depends(get_db)
 ):
-    user = UserCreate(username=username, email=email, password=password, confirm_password=confirm_password)
-    registered_user = register(user, db)
-    return RedirectResponse("/registration_successful")
+    if password != confirm_password:
+        # Return to the registration page with an error
+        return templates.TemplateResponse("register.html", {
+            "request": request,
+            "error_message": "Passwords do not match."
+        })
+
+    try:
+        user = UserCreate(username=username, email=email, password=password, confirm_password=confirm_password)
+        register(user, db)  # If this function raises an HTTPException, it should be handled
+    except HTTPException as e:
+        # Return to the registration page with the error detail
+        return templates.TemplateResponse("register.html", {
+            "request": request,
+            "error_message": e.detail
+        })
+
+    # Redirect to the successful registration page
+    response = RedirectResponse("/registration_successful", status_code=status.HTTP_302_FOUND)
+    return response
     
     
 @app.post("/registration_successful", response_class=HTMLResponse)