Update utils/suggestions.py

utils/suggestions.py

@@ -2,10 +2,8 @@ import requests
 import os
 
 HF_API_KEY = os.getenv("HF_API_KEY")
-ADZUNA_API_KEY = os.getenv("ADZUNA_API_KEY")
-ADZUNA_APP_ID = os.getenv("ADZUNA_APP_ID")
-
-HF_API_URL = "https://api-inference.huggingface.co/models/bigscience/T0pp"
+HF_MODEL = "mrm8488/t5-base-finetuned-question-generation-ap"
+HF_API_URL = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
 
 HEADERS = {
     "Authorization": f"Bearer {HF_API_KEY}",
@@ -13,45 +11,49 @@ HEADERS = {
 }
 
 def call_llm(prompt: str) -> str:
-    payload = {"inputs": prompt}
-
-    response = requests.post(HF_API_URL, headers=HEADERS, json=payload)
-
+    response = requests.post(
+        HF_API_URL,
+        headers=HEADERS,
+        json={"inputs": prompt}
+    )
     if response.status_code != 200:
         raise Exception(f"Hugging Face API Error {response.status_code}: {response.text}")
 
     try:
         return response.json()[0]["generated_text"]
-    except Exception:
+    except:
         raise Exception(f"Error parsing Hugging Face response: {response.text}")
 
+
 def get_certification_suggestions(text: str) -> str:
-    prompt = f"Suggest 5 professional certifications based on this CV:\n\n{text}"
+    prompt = f"Suggest 5 professional certifications for this profile:\n{text}"
     return call_llm(prompt)
 
 def get_higher_education_suggestions(text: str) -> str:
-    prompt = f"Suggest suitable higher education programs based on this CV:\n\n{text}"
+    prompt = f"Suggest suitable higher education programs for this profile:\n{text}"
     return call_llm(prompt)
 
 def get_visa_recommendations(text: str) -> str:
-    prompt = f"Suggest visa opportunities in top countries based on this CV:\n\n{text}"
+    prompt = f"Suggest visa options in top countries for this CV:\n{text}"
     return call_llm(prompt)
 
 def get_career_advice(text: str) -> str:
-    prompt = f"Act like a career counselor. Give personalized advice for this CV:\n\n{text}"
+    prompt = f"Give career advice like a counselor for the following profile:\n{text}"
     return call_llm(prompt)
 
+
 def get_job_listings(text: str):
+    ADZUNA_API_KEY = os.getenv("ADZUNA_API_KEY")
+    ADZUNA_APP_ID = os.getenv("ADZUNA_APP_ID")
+
     if not ADZUNA_API_KEY or not ADZUNA_APP_ID:
-        return ["Adzuna API credentials not found in environment variables."]
-    
-    keyword = "engineer" if "engineer" in text.lower() else "professional"
+        return ["Adzuna API credentials missing."]
 
     params = {
         "app_id": ADZUNA_APP_ID,
         "app_key": ADZUNA_API_KEY,
-        "what": keyword,
-        "results_per_page": 5,
+        "what": "engineer",
+        "results_per_page": 5
     }
 
     url = "https://api.adzuna.com/v1/api/jobs/us/search/1"
@@ -59,9 +61,8 @@ def get_job_listings(text: str):
 
     if response.status_code == 200:
         try:
-            data = response.json()
-            return [f"{job['title']} - {job['location']['display_name']}" for job in data["results"]]
+            jobs = response.json()["results"]
+            return [f"{j['title']} - {j['location']['display_name']}" for j in jobs]
         except:
-            return ["Error parsing job data."]
-    else:
-        return [f"Job API Error {response.status_code}: {response.text}"]
+            return ["Error parsing Adzuna job data."]
+    return [f"Job API Error {response.status_code}: {response.text}"]