Update main.py

main.py

@@ -32,6 +32,7 @@ root_path = os.getenv("HF_SPACE_ROOT_PATH", "")
 class UserProfile(BaseModel):
     skills: List[str] = Field(..., example=["python", "data analysis"])
     sectors: List[str] = Field(..., example=["machine learning", "web development"])
+    internshipType: str = Field(..., example="Bengaluru")
 
 class SearchQuery(BaseModel):
     query: str = Field(..., example="marketing internship in mumbai")
@@ -181,12 +182,12 @@ def get_profile_recommendations(profile: UserProfile):
     if chroma_collection is None or encoder is None:
         raise HTTPException(status_code=503, detail="Server is not ready.")
 
-    query_text = f"Skills: {', '.join(profile.skills)}. Sectors: {', '.join(profile.sectors)}"
+    query_text = f"Skills: {', '.join(profile.skills)}. Sectors: {', '.join(profile.sectors)}. internshipType: {profile.location}"
     query_embedding = encoder.encode([query_text])[0].tolist()
     
     results = chroma_collection.query(
         query_embeddings=[query_embedding], 
-        n_results=random.randint(3, 5) # Get 3 to 5 results
+        n_results=random.randint(5, 7) # Get 5 to 7 results
     )
     
     recommendations = []