Bug fixes for uploading docx file

.env

@@ -4,7 +4,7 @@ ENVIRONMENT=dev
 DB_HOST=localhost
 DB_USER=postgres
 DB_PORT=5432
-DB_PASSWORD=admin
+DB_PASSWORD=quick
 DB_NAME=QuickGpt
 
 SUPER_ADMIN_EMAIL=superadmin@email.com
@@ -15,9 +15,8 @@ SECRET_KEY=ba9dc3f976cf8fb40519dcd152a8d7d21c0b7861d841711cdb2602be8e85fd7c
 ACCESS_TOKEN_EXPIRE_MINUTES=60                
 REFRESH_TOKEN_EXPIRE_MINUTES = 120 # 7 days
 
-
-SMTP_SERVER=mail.gibl.com.np
-SMTP_PORT=25
-SMTP_SENDER_EMAIL=noreply@gibl.com.np
-SMTP_USERNAME=noreply@gibl.com.np
-SMTP_PASSWORD=*G15y^N0reP!y
+SMTP_SERVER=smtp.gmail.com
+SMTP_PORT=587
+SMTP_SENDER_EMAIL=shresthasaurab030@outlook.com
+SMTP_USERNAME=shresthasaurab030
+SMTP_PASSWORD=huurxwxeorxjorzw

private_gpt/__main__.py

@@ -5,10 +5,11 @@ import uvicorn
 from private_gpt.main import app
 from private_gpt.settings.settings import settings
 from fastapi.staticfiles import StaticFiles
+from private_gpt.constants import UPLOAD_DIR
 
 # Set log_config=None to do not use the uvicorn logging configuration, and
 # use ours instead. For reference, see below:
 # https://github.com/tiangolo/fastapi/discussions/7457#discussioncomment-5141108
-app.mount("/static", StaticFiles(directory=r"C:\Users\Dbuser\QuickGPT\backend\privateGPT\private_gpt\static"), name="static")
+app.mount("/static", StaticFiles(directory=UPLOAD_DIR), name="static")
 
 uvicorn.run(app, host="0.0.0.0", port=settings().server.port, log_config=None)

private_gpt/constants.py

@@ -2,4 +2,5 @@ import os
 from pathlib import Path
 
 PROJECT_ROOT_PATH: Path = Path(__file__).parents[1]
-UPLOAD_DIR = r"C:\Users\Dbuser\QuickGPT\backend\privateGPT\private_gpt\static"
+script_dir = os.path.dirname(os.path.abspath(__file__))
+UPLOAD_DIR = os.path.join(script_dir, "static")

private_gpt/users/api/v1/routers/auth.py

@@ -28,7 +28,12 @@ def register_user(
     """
     print(f"{email} {fullname} {password} {company.id}")
     user_in = schemas.UserCreate(email=email, password=password, fullname=fullname, company_id=company.id)
-    send_registration_email(fullname, email, password)
+    try:
+        send_registration_email(fullname, email, password)
+    except Exception as e:
+        print(f"Failed to send registration email: {str(e)}")
+        raise HTTPException(
+            status_code=500, detail=f"Failed to send registration email.")
     return crud.user.create(db, obj_in=user_in)
 
 
@@ -159,7 +164,7 @@ def register(
     if existing_user:
         raise HTTPException(
             status_code=409,
-            detail="The user with this email already exists in the system",
+            detail="The user with this email already exists!",
         )
     random_password = security.generate_random_password()
 
@@ -174,7 +179,7 @@ def register(
         if current_user.user_role.role.name not in {Role.SUPER_ADMIN["name"], Role.ADMIN["name"]}:
             raise HTTPException(
                 status_code=403,
-                detail="You do not have permission to register users for a specific company.",
+                detail="You do not have permission to register users!",
             )
         user = register_user(db, email, fullname, random_password, company)
         user_role_name = role_name or Role.GUEST["name"]

private_gpt/users/api/v1/routers/users.py

@@ -31,10 +31,10 @@ def read_users(
     return users
 
 
-@router.get("/company/{company_name}", response_model=List[schemas.User])
+@router.get("/company/{company_id}", response_model=List[schemas.User])
 def read_users_by_company(
-    company_name: Optional[str] = Path(..., title="Company Name",
-                                       description="Only for company admin"),
+    company_id: int = Path(..., title="Company ID",
+                           description="Only for company admin"),
     db: Session = Depends(deps.get_db),
     current_user: models.User = Security(
         deps.get_current_user,
@@ -42,12 +42,12 @@ def read_users_by_company(
     ),
 ):
     """Retrieve all users of that company only"""
-    company = crud.company.get_by_company_name(db, company_name=company_name)
+    company = crud.company.get(db, company_id)
 
     if company is None:
         raise HTTPException(
             status_code=status.HTTP_404_NOT_FOUND,
-            detail=f"Company with name '{company_name}' not found",
+            detail=f"Company with ID '{company_id}' not found",
         )
 
     users = crud.user.get_multi_by_company_id(db, company_id=company.id)
@@ -262,11 +262,11 @@ def admin_change_password(
     )
 
 
-@router.delete("/{user_id}")
+@router.post("/delete")
 def delete_user(
     *,
     db: Session = Depends(deps.get_db),
-    user_id: int,
+    delete_user: schemas.DeleteUser,
     current_user: models.User = Security(
         deps.get_current_user,
         scopes=[Role.ADMIN["name"], Role.SUPER_ADMIN["name"]],
@@ -275,6 +275,7 @@ def delete_user(
     """
     Delete a user by ID.
     """
+    user_id = delete_user.id
     user = crud.user.get(db, id=user_id)
     if user is None:
         raise HTTPException(status_code=404, detail="User not found")
@@ -283,3 +284,4 @@ def delete_user(
         status_code=status.HTTP_200_OK,
         content={"message": "User deleted successfully"},
     )
+

private_gpt/users/crud/user_crud.py

@@ -3,9 +3,11 @@ from typing import Any, Dict, List, Optional, Union
 from private_gpt.users.core.security import get_password_hash, verify_password
 from private_gpt.users.crud.base import CRUDBase
 from private_gpt.users.models.user import User
-from private_gpt.users.schemas.user import UserCreate, UserUpdate
+from private_gpt.users.schemas.user import UserCreate, UserUpdate, AdminUpdate
+from private_gpt.users.models.user_role import UserRole
+from private_gpt.users.models.role import Role
 from sqlalchemy.orm import Session
-
+from sqlalchemy.orm import joinedload
 
 class CRUDUser(CRUDBase[User, UserCreate, UserUpdate]):
     def get_by_email(self, db: Session, *, email: str) -> Optional[User]:
@@ -27,7 +29,7 @@ class CRUDUser(CRUDBase[User, UserCreate, UserUpdate]):
         db: Session,
         *,
         db_obj: User,
-        obj_in: Union[UserUpdate, Dict[str, Any]],
+        obj_in: Union[AdminUpdate, Dict[str, Any]],
     ) -> User:
         if isinstance(obj_in, dict):
             update_data = obj_in
@@ -38,7 +40,7 @@ class CRUDUser(CRUDBase[User, UserCreate, UserUpdate]):
             del update_data["password"]
             update_data["hashed_password"] = hashed_password
         return super().update(db, db_obj=db_obj, obj_in=update_data)
-
+    
     def get_multi(
         self, db: Session, *, skip: int = 0, limit: int = 100,
     ) -> List[User]:
@@ -73,15 +75,19 @@ class CRUDUser(CRUDBase[User, UserCreate, UserUpdate]):
             .all()
         )
 
+
     def get_multi_by_company_id(
         self, db: Session, *, company_id: str, skip: int = 0, limit: int = 100
     ) -> List[User]:
         return (
             db.query(self.model)
-            .filter(User.company_id == company_id)
+            .join(User.user_role)
+            .filter(UserRole.company_id == company_id)
+            .options(joinedload(User.user_role).joinedload(UserRole.role))
             .offset(skip)
             .limit(limit)
             .all()
         )
+    
 
 user = CRUDUser(User)

private_gpt/users/crud/user_role_crud.py

@@ -17,4 +17,9 @@ class CRUDUserRole(CRUDBase[UserRole, UserRoleCreate, UserRoleUpdate]):
     )-> Optional[UserRole]:
         return db.query(UserRole).filter(UserRole.user_id == user_id).delete()
 
+    # def update_user_role(
+    #         self, db: Session, *, user_id: int, role_id: int
+    # ) -> Optional[UserRole]:
+    #     return self.update(db, )
+    
 user_role = CRUDUserRole(UserRole)

private_gpt/users/schemas/__init__.py

@@ -1,6 +1,6 @@
 from .role import Role, RoleCreate, RoleInDB, RoleUpdate
 from .token import TokenSchema, TokenPayload
-from .user import User, UserCreate, UserInDB, UserUpdate, UserBaseSchema, Profile, UsernameUpdate
+from .user import User, UserCreate, UserInDB, UserUpdate, UserBaseSchema, Profile, UsernameUpdate, DeleteUser, AdminUpdate
 from .user_role import UserRole, UserRoleCreate, UserRoleInDB, UserRoleUpdate
 from .subscription import Subscription, SubscriptionBase, SubscriptionCreate, SubscriptionUpdate
 from .company import Company, CompanyBase, CompanyCreate, CompanyUpdate

private_gpt/users/schemas/user.py

@@ -53,4 +53,11 @@ class UserInDB(UserSchema):
 
 
 class Profile(UserBaseSchema):
-	role: str
+	role: str
+
+class DeleteUser(BaseModel):
+	id: int
+
+class AdminUpdate(BaseModel):
+	fullname: str
+	role: int

private_gpt/users/utils/utils.py

@@ -9,12 +9,24 @@ def send_registration_email(fullname: str, email: str, random_password: str) ->
     Send a registration email with a random password.
     """
     subject = "Welcome to QuickGPT - Registration Successful"
-    body = f"Hello {fullname},\n\nThank you for registering with QuickGPT!\n\n"\
-       f"Your temporary password is: {random_password}\n"\
-       f"Please login to the QuickGPT: http://quickgpt.gibl.com.np\n"\
-       f"Please use this password to log in and consider changing it"\
-       " to a more secure one after logging in.\n\n"\
-       "Best regards,\nQuickGPT Team"
+    body = f"""
+        <html>
+        <body>
+            <p>Hello {fullname},</p>
+            
+            <p>Thank you for registering with QuickGPT!</p>
+            
+            <p>Your temporary password is: <strong>{random_password}</strong></p>
+            
+            <p>Please log in to QuickGPT <a href="http://quickgpt.gibl.com.np">here</a>.</p>
+            
+            <p>Please use this password to log in and consider changing it to a more secure one after logging in.</p>
+            
+            <p>Best regards,<br>
+            QuickGPT Team</p>
+        </body>
+        </html>
+    """
 
     msg = MIMEMultipart()
     msg.attach(MIMEText(body, "plain"))
@@ -26,7 +38,7 @@ def send_registration_email(fullname: str, email: str, random_password: str) ->
     print(settings.SMTP_PORT)
 
     with smtplib.SMTP(settings.SMTP_SERVER, settings.SMTP_PORT) as server:
-        # server.starttls()
-        # server.login(settings.SMTP_USERNAME, settings.SMTP_PASSWORD)
+        server.starttls()
+        server.login(settings.SMTP_USERNAME, settings.SMTP_PASSWORD)
         server.sendmail(settings.SMTP_SENDER_EMAIL, email, msg.as_string())