Update modules/app.py

modules/app.py

@@ -10,13 +10,6 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import HTMLResponse
 import uvicorn
 
-from pydantic import BaseModel
-from pymongo import MongoClient
-import jwt
-from jwt import encode as jwt_encode
-from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
-from bson import ObjectId
-
 import ctranslate2
 import sentencepiece as spm
 import fasttext
@@ -42,83 +35,6 @@ fasttext.FastText.eprint = lambda x: None
 # User interface
 templates_folder = os.path.join(os.path.dirname(__file__), "templates")
 
-# Authentication
-class User(BaseModel):
- username: str = None # Make the username field optional
- email: str
- password: str
-
-# Connect to the MongoDB database
-client = MongoClient("mongodb://localhost:27017")
-db = client["mydatabase"]
-users_collection = db["users"]
-
-# Secret key for signing the token
-SECRET_KEY = "helloworld"
-security = HTTPBearer()
-
-#Implement the login route:
-
-@app.post("/login")
-def login(user: User):
- # Check if user exists in the database
- user_data = users_collection.find_one(
- {"email": user.email, "password": user.password}
- )
- if user_data:
-     # Generate a token
-     token = generate_token(user.email)
-     # Convert ObjectId to string
-     user_data["_id"] = str(user_data["_id"])
-     # Store user details and token in local storage
-     user_data["token"] = token
-     return user_data
- return {"message": "Invalid email or password"}
-
-#Implement the registration route:
-@app.post("/register")
-def register(user: User):
- # Check if user already exists in the database
- existing_user = users_collection.find_one({"email": user.email})
- if existing_user:
-     return {"message": "User already exists"}
- #Insert the new user into the database
- user_dict = user.dict()
- users_collection.insert_one(user_dict)
- # Generate a token
- token = generate_token(user.email)
- # Convert ObjectId to string
- user_dict["_id"] = str(user_dict["_id"])
- # Store user details and token in local storage
- user_dict["token"] = token
- return user_dict
-
-
-#Implement the `/api/user` route to fetch user data based on the JWT token
-@app.get("/api/user")
-def get_user(credentials: HTTPAuthorizationCredentials = Depends(security)):
- # Extract the token from the Authorization header
- token = credentials.credentials
- # Authenticate and retrieve the user data from the database based on the token
- # Here, you would implement the authentication logic and fetch user details
- # based on the token from the database or any other authentication mechanism
- # For demonstration purposes, assuming the user data is stored in local storage
- # Note: Local storage is not accessible from server-side code
- # This is just a placeholder to demonstrate the concept
- user_data = {
- "username": "John Doe",
- "email": "[email protected]"
- }
-if user_data["username"] and user_data["email"]:
- return user_data
-raise HTTPException(status_code=401, detail="Invalid token")
-
-#Define a helper function to generate a JWT token
-def generate_token(email: str) -> str:
- payload = {"email": email}
- token = jwt_encode(payload, SECRET_KEY, algorithm="HS256")
- return token
-
     
 # Get time of request