Update translator.py

translator.py

@@ -1,11 +1,11 @@
 import requests
 import os
 
-# Retrieve Hugging Face API token from environment variable
-API_TOKEN = os.getenv("HF_API_TOKEN")
+# Set Hugging Face API Token
+API_TOKEN = os.getenv("HF_API_TOKEN")  # Get token from environment
 
 if not API_TOKEN:
-    raise ValueError("⚠️ Error: Hugging Face API token is missing! Set the HF_API_TOKEN environment variable.")
+    raise ValueError("⚠️ Error: Hugging Face API token is missing! Set HF_API_TOKEN.")
 
 # Define model and API endpoint
 MODEL_ID = "bigcode/starcoder"
@@ -16,19 +16,32 @@ def translate_code(code_snippet, source_lang, target_lang):
     """
     Translate code using Hugging Face API.
     """
-    prompt = f"Translate the following {source_lang} code to {target_lang}:\n\n{code_snippet}\n\nTranslated {target_lang} Code:\n"
+    prompt = f"""
+    Translate the following {source_lang} code to {target_lang}:
+    
+    {code_snippet}
+    
+    Translated {target_lang} Code:
+    """
 
     try:
         response = requests.post(API_URL, headers=HEADERS, json={"inputs": prompt})
 
+        # Check response status
         if response.status_code == 200:
             response_json = response.json()
+            
+            # Ensure proper response format
             if isinstance(response_json, list) and len(response_json) > 0:
                 generated_text = response_json[0].get("generated_text", "")
-                translated_code = generated_text.split(f"Translated {target_lang} Code:\n")[-1].strip()
-                return translated_code
+                if generated_text:
+                    # Extract translated code
+                    translated_code = generated_text.split(f"Translated {target_lang} Code:")[-1].strip()
+                    return translated_code
+                else:
+                    return "⚠️ Error: No translated text received."
             else:
-                return "⚠️ Error: Unexpected response format from API."
+                return "⚠️ Error: Unexpected response format."
 
         elif response.status_code == 400:
             return "⚠️ Error: Invalid request. Check input format."
@@ -40,7 +53,7 @@ def translate_code(code_snippet, source_lang, target_lang):
             return "⚠️ Error: Access forbidden. You may need special access to this model."
 
         elif response.status_code == 503:
-            return "⚠️ Error: Model is loading. Please wait a moment and try again."
+            return "⚠️ Error: Model is loading. Please wait and try again."
 
         else:
             return f"⚠️ Error {response.status_code}: {response.text}"