Update app.py

app.py

@@ -56,7 +56,14 @@ elif selected == 'Translate':
                 response = requests.post(API_TRANSLATED, 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)
+            if not output_translate[0]["translation_text"]:
+                    error_message = output_translate[0]["error"]
+                    st.error(f"Le texte n'a pas pu être traduit: {error_message}")
+                else:
+                    translated_text = output_translate[0]["translation_text"]
+                    col5.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ translated_text +'''</p>''',  unsafe_allow_html=True)
+                
+            # text_out_spanish = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output_spanish[0] ["translation_text"] +'''</p>''',  unsafe_allow_html=True)
             st.info(":blue[Please feel fit to like it after use! 😉🙏]")
         elif languages == "english - french" and text_in:
             API_URL_french = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-fr"
@@ -83,7 +90,14 @@ elif selected == 'Translate':
                     response = requests.post(API_TRANSLATED, 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)
+                if not output_translate[0]["translation_text"]:
+                    error_message = output_translate[0]["error"]
+                    st.error(f"Le texte n'a pas pu être traduit: {error_message}")
+                else:
+                    translated_text = output_translate[0]["translation_text"]
+                    col5.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ translated_text +'''</p>''',  unsafe_allow_html=True)
+                
+                # text_out_chinese = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output_chinese[0] ["translation_text"] +'''</p>''',  unsafe_allow_html=True)
                 st.info(":blue[Please feel fit to like it after use! 😉🙏]")
         elif languages == "english - german" and text_in:
             headers_german = {"Authorization": API_TOKEN}
@@ -92,7 +106,14 @@ elif selected == 'Translate':
                     response = requests.post(API_TRANSLATED, 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)
+                if not output_translate[0]["translation_text"]:
+                    error_message = output_translate[0]["error"]
+                    st.error(f"Le texte n'a pas pu être traduit: {error_message}")
+                else:
+                    translated_text = output_translate[0]["translation_text"]
+                    col5.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ translated_text +'''</p>''',  unsafe_allow_html=True)
+                
+                # text_out_german = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output_german[0] ["translation_text"] +'''</p>''',  unsafe_allow_html=True)
                 st.info(":blue[Please feel fit to like it after use! 😉🙏]")
 
 elif selected == 'Summarization':