Update main.py

main.py

@@ -1,26 +1,125 @@
 '''
         Created By Lewis Kamau Kimaru 
-        Sema translator api backend
+        Sema translator fastapi implementation
         January 2024
         Docker deployment
 '''
 
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, HTTPException, Request, Depends
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import HTMLResponse
-import gradio as gr
+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
-import uvicorn
+
 import pytz
 from datetime import datetime
 import os
 
 app = FastAPI()
 
+origins = ["*"]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=False,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 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": "johndoe@example.com"
+ }
+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
 
 def get_time():
@@ -111,23 +210,12 @@ def translate_enter(userinput: str, source_lang: str, target_lang: str):
   # Return the source language and the translated text
   return translations_desubword[0]
 
+    
+@app.get("/", response_class=HTMLResponse)
+async def read_root(request: Request):
+    return HTMLResponse(content=open(os.path.join(templates_folder, "translator.html"), "r").read(), status_code=200)
 
-@app.get("/")
-async def read_root():
-    gradio_interface = """
-    <html>
-    <meta name="viewport" content="width=device-width, height=device-height, initial-scale=1.0">
-    <head>
-        <title>Sema</title>
-    </head>
-    <frameset>
-        <frame src=https://kamau1-semaapi-frontend.hf.space/?embedded=true'>
-    </frameset>
-    </html>
-    """
-    return HTMLResponse(content=gradio_interface)
-
-
+    
 @app.post("/translate_detect/")
 async def translate_detect_endpoint(request: Request):
     datad = await request.json()