Update app.py

app.py

@@ -1,20 +1,29 @@
 import streamlit as st 
-import webbrowser
-
+# import webbrowser
+from streamlit.components.v1 import html
 
 st.title('Chat LLMs Interfaces !')
 
 mistral_url = 'https://aipatseer-chatllm.hf.space'
 llama2_url = 'https://huggingface-projects-llama-2-7b-chat.hf.space'
 st.title('Select Interface')
-# st.image('mistral_logo.webp')
-# st.image('llama2_logo.png')
+st.image('mistral_logo.webp')
+st.image('llama2_logo.png')
+
 
 
-if st.button('Open Mistral Chat Interface'):
-    webbrowser.open(mistral_url)
+def open_page(url):
+    open_script= """
+        <script type="text/javascript">
+            window.open('%s', '_blank').focus();
+        </script>
+    """ % (url)
+    html(open_script)
+    
+if st.button('Mistral Chat', on_click=open_page, args=(mistral_url,)):
+    #webbrowser.open(mistral_url)
     # st.markdown(mistral_url, unsafe_allow_html=True)
-elif st.button('Open Llama2 Chat Interface'):
-    webbrowser.open(llama2_url)
+elif st.button('Llama2 Chat', on_click=open_page, args=(llama2_url,)):
+    #webbrowser.open(llama2_url)
     # st.markdown(llama2_url, unsafe_allow_html=True)