MLDeveloper commited on
Commit
3e86e45
·
verified ·
1 Parent(s): 6f47c9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +61 -33
app.py CHANGED
@@ -1,49 +1,77 @@
1
  import streamlit as st
2
  import requests
3
- import os # Import os to access environment variables
4
 
5
- # Get API token from environment variable
6
- API_TOKEN = os.getenv("HF_API_TOKEN") # Fetch token securely
 
 
 
 
 
7
  MODEL_ID = "bigcode/starcoder"
8
  API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
9
  HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
10
 
11
  def translate_code(code_snippet, source_lang, target_lang):
12
- """Translate code using Hugging Face API securely."""
13
- prompt = f"Translate the following {source_lang} code to {target_lang}:\n\n{code_snippet}\n\nTranslated {target_lang} Code:\n"
14
-
15
- response = requests.post(API_URL, headers=HEADERS, json={
16
- "inputs": prompt,
17
- "parameters": {
18
- "max_new_tokens": 150,
19
- "temperature": 0.2,
20
- "top_k": 50,
21
- "stop": ["\n\n", "#", "//", "'''"]
22
- }
23
- })
24
-
25
- if response.status_code == 200:
26
- generated_text = response.json()[0]["generated_text"]
27
- translated_code = generated_text.split(f"Translated {target_lang} Code:\n")[-1].strip()
28
- return translated_code
29
- else:
30
- return f"Error: {response.status_code}, {response.text}"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
 
32
  # Streamlit UI
33
- st.title("🔄 Code Translator using StarCoder")
34
- st.write("Translate code between different programming languages using AI.")
35
 
36
  languages = ["Python", "Java", "C++", "C"]
37
-
38
- source_lang = st.selectbox("Select source language", languages)
39
- target_lang = st.selectbox("Select target language", languages)
40
- code_input = st.text_area("Enter your code here:", height=200)
41
 
42
  if st.button("Translate"):
43
- if code_input.strip():
44
- with st.spinner("Translating..."):
 
 
 
 
45
  translated_code = translate_code(code_input, source_lang, target_lang)
46
- st.subheader("Translated Code:")
47
  st.code(translated_code, language=target_lang.lower())
48
- else:
49
- st.warning("⚠️ Please enter some code before translating.")
 
1
  import streamlit as st
2
  import requests
3
+ import os
4
 
5
+ # Ensure the Hugging Face API Token is available
6
+ API_TOKEN = os.getenv("HF_API_TOKEN")
7
+ if not API_TOKEN:
8
+ st.error("⚠️ API Token is missing! Please set HF_API_TOKEN as an environment variable.")
9
+ st.stop()
10
+
11
+ # Define Hugging Face Model and API Endpoint
12
  MODEL_ID = "bigcode/starcoder"
13
  API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
14
  HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
15
 
16
  def translate_code(code_snippet, source_lang, target_lang):
17
+ """Translate code between languages using Hugging Face API."""
18
+ prompt = f"""You are an expert AI in code translation. Convert the following {source_lang} code to {target_lang}:
19
+
20
+ {source_lang} Code:
21
+ ```{source_lang.lower()}
22
+ {code_snippet}
23
+ ```
24
+
25
+ Now, provide the equivalent {target_lang} code:
26
+ """
27
+
28
+ try:
29
+ response = requests.post(API_URL, headers=HEADERS, json={"inputs": prompt})
30
+
31
+ # Handle API response
32
+ if response.status_code == 200:
33
+ result = response.json()
34
+ if isinstance(result, list) and result:
35
+ generated_text = result[0].get("generated_text", "")
36
+ translated_code = generated_text.split(f"```{target_lang.lower()}")[-1].strip("```").strip()
37
+ return translated_code
38
+ else:
39
+ return "⚠️ Error: Unexpected API response format."
40
+
41
+ elif response.status_code == 400:
42
+ return "⚠️ Error: Bad request. Please check your input."
43
+
44
+ elif response.status_code == 401:
45
+ return "⚠️ Error: Unauthorized. Check your API token."
46
+
47
+ elif response.status_code == 403:
48
+ return "⚠️ Error: Access Forbidden. You may need special model access."
49
+
50
+ elif response.status_code == 503:
51
+ return "⚠️ Error: Model is loading. Please wait a moment and try again."
52
+
53
+ else:
54
+ return f"⚠️ API Error {response.status_code}: {response.text}"
55
+
56
+ except requests.exceptions.RequestException as e:
57
+ return f"⚠️ Network Error: {str(e)}"
58
 
59
  # Streamlit UI
60
+ st.title("🔄 AI Code Translator")
61
+ st.write("Convert code between Python, Java, C++, and C.")
62
 
63
  languages = ["Python", "Java", "C++", "C"]
64
+ source_lang = st.selectbox("Select Source Language", languages)
65
+ target_lang = st.selectbox("Select Target Language", languages)
66
+ code_input = st.text_area("Enter your code:", height=200)
 
67
 
68
  if st.button("Translate"):
69
+ if source_lang == target_lang:
70
+ st.warning("⚠️ Source and target languages must be different!")
71
+ elif not code_input.strip():
72
+ st.warning("⚠️ Please enter some code before translating.")
73
+ else:
74
+ with st.spinner("Translating... ⏳"):
75
  translated_code = translate_code(code_input, source_lang, target_lang)
76
+ st.subheader(f"Translated {target_lang} Code:")
77
  st.code(translated_code, language=target_lang.lower())