Create app.py

app.py

@@ -5,18 +5,17 @@ import os
 
 API_TOKEN = os.environ.get("API_TOKEN")
 
+
 #changement du logo et du titre de mon application en anglais
 st.set_page_config(page_title="TSC", page_icon="books", layout="centered", menu_items=None)
 
 
-
 col1, col2, col3 = st.columns(3)
 xx = col1
 ar = col2.image("keyce.jpg", width=180)
 yy = col3
 
 
-
 # menu horizontal
 selected = option_menu(None, ["Home", "Translate", "Summarization", "Chatbot"], 
     icons=['house', 'globe', "pencil", "chat"], 
@@ -29,7 +28,6 @@ selected = option_menu(None, ["Home", "Translate", "Summarization", "Chatbot"],
     })
 
 
-
 if selected == "Home":
     st.subheader("Welcome to our app !!")
     st.info(":blue[We are the present of tomorrow! Please like it after use!]")
@@ -44,44 +42,44 @@ elif selected == 'Translate':
     #Boutton pour effectuer la traduction
     btn_translate = st.button("🌐 Translate")
     if languages == "english - spanish" and text_in:
-        API_URL = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-es"
-        headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
+        API_URL_spanish = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-es"
+        headers_spanish = {"Authorization": API_TOKEN}
         if btn_translate:
-            def query(payload):
-                response = requests.post(API_URL, headers=headers, json=payload)
+            def query_spanish(payload):
+                response = requests.post(API_URL_spanish, headers=headers_spanish, json=payload)
                 return response.json() 
-            output = query({"inputs": text_in})
-            text_out = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output[0]["translation_text"] +'''</p>''',  unsafe_allow_html=True)
+            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)
             st.info(":blue[Please feel fit to like it after use! 😉🙏]")
     elif languages == "english - french" and text_in:
-        API_URL = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-fr"
-        headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
+        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(payload):
-                response = requests.post(API_URL, headers=headers, json=payload)
+            def query_french(payload):
+                response = requests.post(API_URL_french, headers=headers_french, json=payload)
                 return response.json() 
-            output = query({"inputs": text_in})
-            text_out = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output[0]["translation_text"] +'''</p>''',  unsafe_allow_html=True)
+            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)
             st.info(":blue[Please feel fit to like it after use! 😉🙏]")   
     elif languages == 'english - chinese' and text_in:
-        API_URL = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-zh"
-        headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}       
+        API_URL_chinese = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-zh"
+        headers_chinese = {"Authorization": API_TOKEN}       
         if btn_translate:
-            def query(payload):
-                response = requests.post(API_URL, headers=headers, json=payload)
+            def query_chinese(payload):
+                response = requests.post(API_URL_chinese, headers=headers_chinese, json=payload)
                 return response.json() 
-            output = query({"inputs": text_in})
-            text_out = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output[0]["translation_text"] +'''</p>''',  unsafe_allow_html=True)
+            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)
             st.info(":blue[Please feel fit to like it after use! 😉🙏]")
     elif languages == "english - german" and text_in:
-        API_URL = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-de"
-        headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
+        API_URL_german = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-de"
+        headers_german = {"Authorization": API_TOKEN}
         if btn_translate:
-            def query(payload):
-                response = requests.post(API_URL, headers=headers, json=payload)
+            def query_german(payload):
+                response = requests.post(API_URL_german, headers=headers_german, json=payload)
                 return response.json() 
-            output = query({"inputs": text_in})
-            text_out = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output[0]["translation_text"] +'''</p>''',  unsafe_allow_html=True)
+            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)
             st.info(":blue[Please feel fit to like it after use! 😉🙏]")
 
 elif selected == 'Summarization':
@@ -90,30 +88,36 @@ elif selected == 'Summarization':
     btn_sum = st.button("📝 Summary")
     if text_sum:
         if btn_sum:
-            API_URL = "https://api-inference.huggingface.co/models/Falconsai/text_summarization"
-            headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
-            def query_sum(payload):
-                response = requests.post(API_URL, headers=headers, json=payload)
+            API_URL_summary = "https://api-inference.huggingface.co/models/Falconsai/text_summarization"
+            headers_summary = {"Authorization": API_TOKEN}
+            def query_summary(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_sum({"inputs": text_sum})
-            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)
+            output_sum = query_summary({"inputs": text_sum})
+            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! 😉🙏]")
 
 elif selected == 'Chatbot':
-    texte_chat = st.text_input("Enter your question:")
+    texte_chat = st.text_input("Enter your question")
     #Boutton pour effectuer le chatbot
     btn_chat = st.button("💬 Submit")
-    API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct"
-    headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
+    API_URL_chatbot = "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct"
+    headers_chatbot = {"Authorization": API_TOKEN}
 
     if texte_chat:
         if btn_chat:
-            def chatbot(payload):
-                response = requests.post(API_URL, headers=headers, json=payload)
+            def query_chatbot(payload):
+                response = requests.post(API_URL_chatbot, headers=headers_chatbot, json=payload)
                 return response.json()
-            output_chat = chatbot({"inputs": texte_chat})
+            output_chat = query_chatbot({"inputs": texte_chat})
             text_out_chat = st.info(':green['+ output_chat[0] ["generated_text"] +']')
             st.info(":blue[Please feel fit to like it after use! 😉🙏]")
 
-st.markdown('''<br/> <h6 style='text-align:center;color:grey;font-size:11px;'> © 2024 - TOUNDE - All rights reserved </h6>''',  unsafe_allow_html=True)
+st.markdown('''<br/> <h6 style='text-align:center;color:grey;font-size:11px;'> © 2024 - TOUNDE - All rights reserved </h6>''',  unsafe_allow_html=True)
+
+# for key in range(5):
+#     try:
+#         text_out = col4.markdown('''<br/> <p style='text-align:left;font-size:16px;'>'''+ output[0]["translation_text"] +'''</p>''',  unsafe_allow_html=True)
+#     except KeyError:
+#         print("Couldn't find a match for the key:", key)