Update utils/suggestions.py

utils/suggestions.py

@@ -2,7 +2,7 @@ import requests
 import os
 
 HF_API_KEY = os.getenv("HF_API_KEY")
-HF_MODEL = "mrm8488/t5-base-finetuned-question-generation-ap"
+HF_MODEL = "google/flan-t5-small"
 HF_API_URL = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
 
 HEADERS = {
@@ -10,6 +10,7 @@ HEADERS = {
     "Content-Type": "application/json"
 }
 
+
 def call_llm(prompt: str) -> str:
     response = requests.post(
         HF_API_URL,
@@ -21,24 +22,27 @@ def call_llm(prompt: str) -> str:
 
     try:
         return response.json()[0]["generated_text"]
-    except:
+    except Exception as e:
         raise Exception(f"Error parsing Hugging Face response: {response.text}")
 
 
 def get_certification_suggestions(text: str) -> str:
-    prompt = f"Suggest 5 professional certifications for this profile:\n{text}"
+    prompt = f"Suggest 5 certifications for this resume: {text}"
     return call_llm(prompt)
 
+
 def get_higher_education_suggestions(text: str) -> str:
-    prompt = f"Suggest suitable higher education programs for this profile:\n{text}"
+    prompt = f"Suggest higher education programs for this resume: {text}"
     return call_llm(prompt)
 
+
 def get_visa_recommendations(text: str) -> str:
-    prompt = f"Suggest visa options in top countries for this CV:\n{text}"
+    prompt = f"Suggest visa opportunities in top countries based on this resume: {text}"
     return call_llm(prompt)
 
+
 def get_career_advice(text: str) -> str:
-    prompt = f"Give career advice like a counselor for the following profile:\n{text}"
+    prompt = f"Give career counseling advice based on this resume: {text}"
     return call_llm(prompt)