Update app.py

app.py

@@ -36,7 +36,14 @@ if selected == "Home":
     st.write("Hello to everybody! In this Artificial Intelligence app, we offer you a THREE-IN-ONE package. First of all, we've got the (Translate) menu, which lets you do a good job of translating into a few foreign languages, then we've got the (Summary) option, which lets you enter your text in order to summarize it, and finally, as a bonus, we've integrated a chatbot model in the (Chatbot) option, which lets you try out our chatbot, which we're currently perfecting.") 
 
 elif selected == 'Translate':
-    languages = st.selectbox('Select input and translate languages', ('', 'english - chinese', 'english - french', 'english - german', 'english - spanish'))
+    # Choose the translation language
+    translation_model = {
+        "english - spanish": "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-es",
+        "english - french": "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-fr",
+        "english - chinese": "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-zh",
+        "english - german": "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-de"
+    }
+    languages = st.selectbox('Select input and translate languages', (list(translation_models.keys()))
     col3, col4 = st.columns(2)
     text_in = col3.text_area("", height=200, placeholder='type or paste the text you wish to translate.')
     #Boutton pour effectuer la traduction
@@ -46,7 +53,7 @@ elif selected == 'Translate':
         headers_spanish = {"Authorization": API_TOKEN}
         if btn_translate:
             def query_spanish(payload):
-                response = requests.post(API_URL_spanish, headers=headers_spanish, json=payload)
+                response = requests.post(translation_model, headers=headers_spanish, json=payload)
                 return response.json() 
             output_spanish = query_spanish({"inputs": text_in})
             text_out_spanish = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output_spanish[0] ["translation_text"] +'''</p>''',  unsafe_allow_html=True)
@@ -56,7 +63,7 @@ elif selected == 'Translate':
         headers_french = {"Authorization": API_TOKEN}
         if btn_translate:
             def query_french(payload):
-                response = requests.post(API_URL_french, headers=headers_french, json=payload)
+                response = requests.post(translation_model, headers=headers_french, json=payload)
                 return response.json() 
             output_french = query_french({"inputs": text_in})
             text_out_french = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output_french[0] ["translation_text"] +'''</p>''',  unsafe_allow_html=True)
@@ -66,7 +73,7 @@ elif selected == 'Translate':
         headers_chinese = {"Authorization": API_TOKEN}       
         if btn_translate:
             def query_chinese(payload):
-                response = requests.post(API_URL_chinese, headers=headers_chinese, json=payload)
+                response = requests.post(translation_model, headers=headers_chinese, json=payload)
                 return response.json() 
             output_chinese = query_chinese({"inputs": text_in})
             text_out_chinese = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output_chinese[0] ["translation_text"] +'''</p>''',  unsafe_allow_html=True)
@@ -76,7 +83,7 @@ elif selected == 'Translate':
         headers_german = {"Authorization": API_TOKEN}
         if btn_translate:
             def query_german(payload):
-                response = requests.post(API_URL_german, headers=headers_german, json=payload)
+                response = requests.post(translation_model, headers=headers_german, json=payload)
                 return response.json() 
             output_german = query_german({"inputs": text_in})
             text_out_german = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output_german[0] ["translation_text"] +'''</p>''',  unsafe_allow_html=True)