msg

main.py

@@ -1,5 +1,5 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException, Depends
-from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
+from fastapi import FastAPI, File, UploadFile, HTTPException, Depends, Form
+from fastapi.security import OAuth2PasswordBearer
 from jose import JWTError, jwt
 from pinecone import Pinecone
 import os
@@ -13,7 +13,8 @@ from datetime import datetime, timedelta
 # Load environment variables
 load_dotenv()
 
-# Secret key for JWT
+# JWT Config
+SECRET_KEY = os.getenv("JWT_SECRET", "default_secret")  # Use a secure secret in production
 ALGORITHM = "HS256"
 ACCESS_TOKEN_EXPIRE_MINUTES = 30
 
@@ -41,15 +42,18 @@ unsplash_index = pc.Index(index_name)
 # Load CLIP model and processor
 model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
 processor = AutoProcessor.from_pretrained("openai/clip-vit-base-patch32")
+model.eval()  # Ensure model is in evaluation mode
 
 # OAuth2 authentication
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/token")
 
+
 def create_access_token(data: dict, expires_delta: timedelta = None):
     to_encode = data.copy()
     expire = datetime.utcnow() + (expires_delta or timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES))
     to_encode.update({"exp": expire})
-    return jwt.encode(to_encode, "secret", algorithm=ALGORITHM)
+    return jwt.encode(to_encode, SECRET_KEY, algorithm=ALGORITHM)
+
 
 def authenticate_user(username: str, password: str):
     user = fake_users_db.get(username)
@@ -57,9 +61,10 @@ def authenticate_user(username: str, password: str):
         return None
     return user
 
+
 def get_current_user(token: str = Depends(oauth2_scheme)):
     try:
-        payload = jwt.decode(token, "secret", algorithms=[ALGORITHM])
+        payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
         username: str = payload.get("sub")
         if username is None or username not in fake_users_db:
             raise HTTPException(status_code=401, detail="Invalid authentication")
@@ -67,24 +72,28 @@ def get_current_user(token: str = Depends(oauth2_scheme)):
     except JWTError:
         raise HTTPException(status_code=401, detail="Invalid authentication")
 
+
 @app.post("/token")
-async def login(form_data: OAuth2PasswordRequestForm = Depends()):
-    user = authenticate_user(form_data.username, form_data.password)
+async def login(username: str = Form(...), password: str = Form(...)):
+    user = authenticate_user(username, password)
     if not user:
         raise HTTPException(status_code=400, detail="Incorrect username or password")
     access_token = create_access_token(data={"sub": user["username"]})
     return {"access_token": access_token, "token_type": "bearer"}
 
+
 def get_text_embedding(text: str):
     inputs = processor(text=[text], return_tensors="pt", padding=True, truncation=True)
     text_features = model.get_text_features(**inputs)
     return text_features.detach().cpu().numpy().flatten().tolist()
 
+
 def get_image_embedding(image: Image.Image):
     inputs = processor(images=image, return_tensors="pt")
     image_features = model.get_image_features(**inputs)
     return image_features.detach().cpu().numpy().flatten().tolist()
 
+
 def search_similar_images(embedding: list, top_k: int = 10):
     results = unsplash_index.query(
         vector=embedding,
@@ -94,6 +103,7 @@ def search_similar_images(embedding: list, top_k: int = 10):
     )
     return results["matches"]
 
+
 @app.get("/search/text/")
 async def search_by_text(query: str, user: str = Depends(get_current_user)):
     if not query:
@@ -102,6 +112,7 @@ async def search_by_text(query: str, user: str = Depends(get_current_user)):
     matches = search_similar_images(embedding)
     return {"matches": [{"id": m["id"], "score": m["score"], "url": m["metadata"]["url"]} for m in matches]}
 
+
 @app.post("/search/image/")
 async def search_by_image(file: UploadFile = File(...), user: str = Depends(get_current_user)):
     try:
@@ -113,6 +124,7 @@ async def search_by_image(file: UploadFile = File(...), user: str = Depends(get_
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error processing image: {str(e)}")
 
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)

requirements.txt

@@ -8,3 +8,4 @@ numpy
 pinecone
 python-jose[cryptography]
 python-multipart
+jose