Update app.py

app.py

@@ -52,15 +52,15 @@ elif selected == 'Translate':
     if btn_translate:
         headers_spanish = {"Authorization": API_TOKEN}
         if languages == "english - spanish" and text_in:
-            def query_spanish(payload):
+            def query(payload):
                 response = requests.post(API_TRANSLATED, headers=headers_spanish, json=payload)
                 return response.json() 
-            output_spanish = query_spanish({"inputs": text_in})
-            if not output_translate[0]["translation_text"]:
-                    error_message = output_translate[0]["error"]
+            output = query_spanish({"inputs": text_in})
+            if not output[0]["translation_text"]:
+                    error_message = output[0]["error"]
                     st.error(f"Le texte n'a pas pu être traduit: {error_message}")
                 else:
-                    translated_text = output_translate[0]["translation_text"]
+                    translated_text = output[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)
@@ -69,13 +69,13 @@ elif selected == 'Translate':
             API_URL_french = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-fr"
             headers_french = {"Authorization": API_TOKEN}
             if btn_translate:
-                def query_french(payload):
+                def query(payload):
                     response = requests.post(API_TRANSLATED, headers=headers_french, json=payload)
                     return response.json() 
-                output_french = query_french({"inputs": text_in})
+                output = query({"inputs": text_in})
 
-                if not output_translate[0]["translation_text"]:
-                    error_message = output_translate[0]["error"]
+                if not output[0]["translation_text"]:
+                    error_message = output[0]["error"]
                     st.error(f"Le texte n'a pas pu être traduit: {error_message}")
                 else:
                     translated_text = output_translate[0]["translation_text"]
@@ -86,15 +86,15 @@ elif selected == 'Translate':
         elif languages == 'english - chinese' and text_in:
             headers_chinese = {"Authorization": API_TOKEN}       
             if btn_translate:
-                def query_chinese(payload):
+                def query(payload):
                     response = requests.post(API_TRANSLATED, headers=headers_chinese, json=payload)
                     return response.json() 
-                output_chinese = query_chinese({"inputs": text_in})
-                if not output_translate[0]["translation_text"]:
-                    error_message = output_translate[0]["error"]
+                output = query({"inputs": text_in})
+                if not output[0]["translation_text"]:
+                    error_message = output[0]["error"]
                     st.error(f"Le texte n'a pas pu être traduit: {error_message}")
                 else:
-                    translated_text = output_translate[0]["translation_text"]
+                    translated_text = output[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)
@@ -102,15 +102,15 @@ elif selected == 'Translate':
         elif languages == "english - german" and text_in:
             headers_german = {"Authorization": API_TOKEN}
             if btn_translate:
-                def query_german(payload):
+                def query(payload):
                     response = requests.post(API_TRANSLATED, headers=headers_german, json=payload)
                     return response.json() 
-                output_german = query_german({"inputs": text_in})
-                if not output_translate[0]["translation_text"]:
-                    error_message = output_translate[0]["error"]
+                output = query({"inputs": text_in})
+                if not output[0]["translation_text"]:
+                    error_message = output[0]["error"]
                     st.error(f"Le texte n'a pas pu être traduit: {error_message}")
                 else:
-                    translated_text = output_translate[0]["translation_text"]
+                    translated_text = output[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)
@@ -124,12 +124,12 @@ elif selected == 'Summarization':
         if text_sum:
             API_URL_summary = "https://api-inference.huggingface.co/models/Falconsai/text_summarization"
             headers_summary = {"Authorization": API_TOKEN}
-            def query_summary(payload):
+            def query(payload):
                 response = requests.post(API_URL_summary, headers=headers_summary, json=payload)
                 return response.json()
             st.markdown('''<h4 style='text-align:left;color:black;'> Answer </h4>''',  unsafe_allow_html=True) 
-            output_sum = query_summary({"inputs": text_sum,})
-            summary_text = output_sum[0]["summary_text"]
+            output = query({"inputs": text_sum,})
+            summary_text = output[0]["summary_text"]
             st.success(f"Response: {summary_text}")
             # text_out_sum = st.markdown('''<p style='text-align:left;font-size:16px;color:orange;font-weight:bold'>'''+ output_sum[0]["summary_text"] +'''</p>''',  unsafe_allow_html=True)
             st.info(":blue[Please feel fit to like it after use! 😉🙏]")
@@ -142,11 +142,11 @@ elif selected == 'Chatbot':
     headers_chatbot = {"Authorization": API_TOKEN}
     if btn_chat:
         if text_chat:
-            def query_chatbot(payload):
+            def query(payload):
                 response = requests.post(API_URL_chatbot, headers=headers_chatbot, json=payload)
                 return response.json()
-            output_chat = query_chatbot({"inputs": text_chat,})
-            generated_text = output_chat[0]["generated_text"]
+            output = query({"inputs": text_chat,})
+            generated_text = output[0]["generated_text"]
             st.success(f"Response: {generated_text}")
             # text_out_chat = st.info(':green['+ output_chat[0]["generated_text"] +']')
             st.info(":blue[Please feel fit to like it after use! 😉🙏]")