app.py

import streamlit as st
import requests
import os

# Get API token from environment variable
API_TOKEN = os.getenv("HF_API_TOKEN")  # Ensure you set this in your environment
if not API_TOKEN:
    st.error("⚠️ API Token is missing! Set HF_API_TOKEN in your environment.")
    st.stop()

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 using Hugging Face API with improved prompt."""
    prompt = f"""
    Translate the following {source_lang} code to {target_lang}:
    
    {code_snippet}
    
    Translated {target_lang} Code:
    """

    response = requests.post(API_URL, headers=HEADERS, json={
        "inputs": prompt
    })

    if response.status_code == 200:
        response_json = response.json()

        if isinstance(response_json, list) and len(response_json) > 0:
            generated_text = response_json[0].get("generated_text", "").strip()
            
            if generated_text:
                # Extract only the 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."

    elif response.status_code == 400:
        return "⚠️ Error: Invalid request. Check input format."

    elif response.status_code == 401:
        return "⚠️ Error: Unauthorized. Check your API token."

    elif response.status_code == 403:
        return "⚠️ Error: Access forbidden. You may need special access to this model."

    elif response.status_code == 503:
        return "⚠️ Error: Model is loading. Please wait and try again."

    else:
        return f"⚠️ Error {response.status_code}: {response.text}"

# Streamlit UI
st.title("🔄 Code Translator using StarCoder")
st.write("Translate code between different programming languages using AI.")

languages = ["Python", "Java", "C++", "C"]

source_lang = st.selectbox("Select source language", languages)
target_lang = st.selectbox("Select target language", languages)
code_input = st.text_area("Enter your code here:", height=200)

if st.button("Translate"):
    if code_input.strip():
        with st.spinner("Translating..."):
            translated_code = translate_code(code_input, source_lang, target_lang)
            st.subheader("Translated Code:")
            st.code(translated_code, language=target_lang.lower())
    else:
        st.warning("⚠️ Please enter some code before translating.")