Update utils/suggestions.py

utils/suggestions.py

@@ -1,38 +1,106 @@
-
+import os
 import requests
+
+HF_API_KEY = os.getenv("HF_API_KEY")
+ADZUNA_APP_ID = os.getenv("ADZUNA_APP_ID")
+ADZUNA_API_KEY = os.getenv("ADZUNA_API_KEY")
+
+def call_llm(prompt: str) -> str:
+    headers = {
+        "Authorization": f"Bearer {HF_API_KEY}",
+        "Content-Type": "application/json"
+    }
+    json_data = {
+        "inputs": prompt,
+        "parameters": {"max_new_tokens": 256},
+    }
+    # Change model URL if needed
+    url = "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct"
+    response = requests.post(url, headers=headers, json=json_data)
+    if response.status_code != 200:
+        raise Exception(f"Hugging Face API Error {response.status_code}: {response.text}")
+    return response.json()[0]['generated_text']
+
+def get_certification_suggestions(text: str) -> str:
+    prompt = f"Suggest professional certifications based on this CV: {text}"
+    return call_llm(prompt)
+
+def get_higher_education_suggestions(text: str) -> str:
+    prompt = f"Suggest suitable higher education programs based on this CV: {text}"
+    return call_llm(prompt)
+
+def get_visa_recommendations(text: str) -> str:
+    prompt = f"Suggest visa opportunities in top countries based on this CV: {text}"
+    return call_llm(prompt)
+
+def get_career_advice(text: str) -> str:
+    prompt = f"Act like a career counselor. Give detailed personalized advice for this CV: {text}"
+    return call_llm(prompt)
+
+def get_job_listings(text: str):
+    headers = {
+        "X-RapidAPI-Key": ADZUNA_API_KEY
+    }
+    params = {
+        "app_id": ADZUNA_APP_ID,
+        "what": "engineer",
+        "results_per_page": 5,
+    }
+    url = "https://api.adzuna.com/v1/api/jobs/us/search/1"
+    res = requests.get(url, headers=headers, params=params)
+    if res.status_code == 200:
+        return [f"{job['title']} - {job['location']['display_name']}" for job in res.json()["results"]]
+    return ["No jobs found"]
 import os
+import requests
 
-HF_API_TOKEN = os.getenv("HF_API_TOKEN")
+HF_API_KEY = os.getenv("HF_API_KEY")
 ADZUNA_APP_ID = os.getenv("ADZUNA_APP_ID")
 ADZUNA_API_KEY = os.getenv("ADZUNA_API_KEY")
 
-def call_llm(prompt):
-    headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
+def call_llm(prompt: str) -> str:
+    headers = {
+        "Authorization": f"Bearer {HF_API_KEY}",
+        "Content-Type": "application/json"
+    }
     json_data = {
         "inputs": prompt,
-        "parameters": {"max_new_tokens": 300},
+        "parameters": {"max_new_tokens": 256},
     }
-    response = requests.post("https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1", headers=headers, json=json_data)
-    if response.status_code == 200:
-        return response.json()[0]["generated_text"]
-    raise Exception(f"Hugging Face API Error {response.status_code}: {response.text}")
+    # Change model URL if needed
+    url = "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct"
+    response = requests.post(url, headers=headers, json=json_data)
+    if response.status_code != 200:
+        raise Exception(f"Hugging Face API Error {response.status_code}: {response.text}")
+    return response.json()[0]['generated_text']
 
-def get_certification_suggestions(text):
-    return call_llm(f"Suggest professional certifications based on this CV: {text}")
+def get_certification_suggestions(text: str) -> str:
+    prompt = f"Suggest professional certifications based on this CV: {text}"
+    return call_llm(prompt)
 
-def get_higher_education_suggestions(text):
-    return call_llm(f"Suggest higher education programs based on this CV: {text}")
+def get_higher_education_suggestions(text: str) -> str:
+    prompt = f"Suggest suitable higher education programs based on this CV: {text}"
+    return call_llm(prompt)
 
-def get_visa_recommendations(text):
-    return call_llm(f"Suggest visa opportunities in top countries based on this CV: {text}")
+def get_visa_recommendations(text: str) -> str:
+    prompt = f"Suggest visa opportunities in top countries based on this CV: {text}"
+    return call_llm(prompt)
 
-def get_career_advice(text):
-    return call_llm(f"Act like a career counselor. Give personalized advice for this CV: {text}")
+def get_career_advice(text: str) -> str:
+    prompt = f"Act like a career counselor. Give detailed personalized advice for this CV: {text}"
+    return call_llm(prompt)
 
-def get_job_listings(text):
+def get_job_listings(text: str):
+    headers = {
+        "X-RapidAPI-Key": ADZUNA_API_KEY
+    }
+    params = {
+        "app_id": ADZUNA_APP_ID,
+        "what": "engineer",
+        "results_per_page": 5,
+    }
     url = "https://api.adzuna.com/v1/api/jobs/us/search/1"
-    params = {"app_id": ADZUNA_APP_ID, "app_key": ADZUNA_API_KEY, "what": "engineer", "results_per_page": 5}
-    response = requests.get(url, params=params)
-    if response.status_code == 200:
-        return [f"{job['title']} - {job['location']['display_name']}" for job in response.json().get("results", [])]
+    res = requests.get(url, headers=headers, params=params)
+    if res.status_code == 200:
+        return [f"{job['title']} - {job['location']['display_name']}" for job in res.json()["results"]]
     return ["No jobs found"]