Update app.py

app.py

@@ -8,20 +8,21 @@ if not API_TOKEN:
     st.error("⚠️ API Token is missing! Please set HF_API_TOKEN as an environment variable.")
     st.stop()
 
-# Use Code Llama for better translation
-MODEL_ID = "codellama/CodeLlama-7b-Instruct-hf"
+# Use StarCoder for better translation
+MODEL_ID = "bigcode/starcoder"
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
 HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
 
 def translate_code(code_snippet, source_lang, target_lang):
     """Translate code between languages using Hugging Face API."""
-    prompt = f"""### Task: Convert {source_lang} code to {target_lang}.
-    
+    prompt = f"""
+    ### Task: Convert {source_lang} code to {target_lang}.
+
     #### {source_lang} Code:
     ```{source_lang.lower()}
     {code_snippet}
     ```
-    
+
     #### Translated {target_lang} Code:
     """
 
@@ -32,10 +33,17 @@ def translate_code(code_snippet, source_lang, target_lang):
             result = response.json()
             if isinstance(result, list) and result:
                 generated_text = result[0].get("generated_text", "")
-                translated_code = generated_text.split(f"#### Translated {target_lang} Code:")[-1].strip()
+                
+                # Extract translated code
+                if f"#### Translated {target_lang} Code:" in generated_text:
+                    translated_code = generated_text.split(f"#### Translated {target_lang} Code:")[-1].strip()
+                else:
+                    translated_code = generated_text.strip()
+
                 return translated_code if translated_code else "⚠️ No translated code received."
-            else:
-                return "⚠️ Unexpected API response format."
+            
+            return "⚠️ Unexpected API response format."
+        
         elif response.status_code == 400:
             return "⚠️ Error: Bad request. Check your input."
         elif response.status_code == 401:
@@ -46,6 +54,7 @@ def translate_code(code_snippet, source_lang, target_lang):
             return "⚠️ Error: Model is loading. Please wait and try again."
         else:
             return f"⚠️ API Error {response.status_code}: {response.text}"
+
     except requests.exceptions.RequestException as e:
         return f"⚠️ Network Error: {str(e)}"