Update utils.py

utils.py

@@ -1,21 +1,24 @@
 import os
 import requests
+import tempfile
 from transformers import pipeline
 
-# Set Hugging Face cache directory to avoid /.cache permission error
-os.environ["HF_HOME"] = "./hf_cache"
-os.makedirs("./hf_cache", exist_ok=True)
+# Use a safe writable temporary directory
+cache_dir = os.path.join(tempfile.gettempdir(), "hf_cache")
+os.environ["HF_HOME"] = cache_dir
+os.makedirs(cache_dir, exist_ok=True)
 
-# Load a text generation model like GPT-2 safely
+# Load GPT-2 safely into tmp directory
 try:
     generator = pipeline(
         "text-generation",
         model="gpt2",
-        cache_dir="./hf_cache"
+        cache_dir=cache_dir,
+        local_files_only=False
     )
 except Exception as e:
     generator = None
-    print("⚠️ Failed to load GPT-2 model:", e)
+    print("⚠️ GPT-2 model loading failed:", e)
 
 
 def generate_description(country_name):
@@ -23,22 +26,19 @@ def generate_description(country_name):
     Generate a simple English description about a country using GPT-2.
     """
     if not generator:
-        return "⚠️ Model is not available. Please check the server logs."
+        return "⚠️ Model unavailable."
 
     try:
         prompt = f"Tell me about {country_name}."
         result = generator(prompt, max_length=100, do_sample=True)
-        if result and isinstance(result, list):
-            return result[0]["generated_text"].strip()
-        else:
-            return "⚠️ Failed to generate text."
+        return result[0]["generated_text"].strip()
     except Exception as e:
-        return f"⚠️ Error generating description: {str(e)}"
+        return f"⚠️ Error: {str(e)}"
 
 
 def get_country_info(country_name):
     """
-    Fetch country data from restcountries.com API.
+    Get basic country info using REST Countries API.
     """
     try:
         url = f"https://restcountries.com/v3.1/name/{country_name}"
@@ -53,9 +53,8 @@ def get_country_info(country_name):
             "Languages": ", ".join(data.get("languages", {}).values())
         }
         return info
-
     except Exception as e:
-        return {"error": f"⚠️ Could not retrieve data for {country_name}."}
+        return {"error": f"⚠️ Failed to retrieve country data."}
 
 
 def translate_to_bangla(text):
@@ -74,5 +73,5 @@ def translate_to_bangla(text):
             }
         )
         return response.json().get("translatedText", "⚠️ Translation failed.")
-    except Exception as e:
+    except Exception:
         return "⚠️ Translation API error."