Update app.py

app.py

@@ -4,15 +4,17 @@ import streamlit as st
 import webchat
 import utils
 
-# URL of the hosted LLMs is hardcoded because at this time all LLMs share the same endpoint
-url = "https://us-south.ml.cloud.ibm.com"
+# Default URL of the hosted LLMs
+default_url = "https://us-south.ml.cloud.ibm.com"
 
 def main():
-    # Initialize session state for credentials
+    # Initialize session state for credentials and url
     if 'watsonx_project_id' not in st.session_state:
         st.session_state['watsonx_project_id'] = ""
     if 'api_key' not in st.session_state:
         st.session_state['api_key'] = ""
+    if 'watsonx_url' not in st.session_state:
+        st.session_state['watsonx_url'] = default_url
 
     st.set_page_config(layout="wide", page_title="RAG Web Demo", page_icon="")
     utils.load_css("styles.css")
@@ -30,16 +32,19 @@ def main():
     st.sidebar.markdown("Insert your credentials of [IBM Cloud](https://cloud.ibm.com/login) for watsonx.ai \n The data is not saved in the server. Your data is secured.", unsafe_allow_html=True)
     st.sidebar.markdown("<hr>", unsafe_allow_html=True)
     
-    # Input fields for API Key and Project ID
+    # Input fields for API Key, Project ID, and Watsonx URL
     api_key_input = st.sidebar.text_input("API Key", st.session_state['api_key'], type="password")
     project_id_input = st.sidebar.text_input("Project ID", st.session_state['watsonx_project_id'])
-    
+    watsonx_url_input = st.sidebar.text_input("Watsonx URL", st.session_state['watsonx_url'])
+
     # Update session state with the provided credentials
     if api_key_input:
         st.session_state['api_key'] = api_key_input
     if project_id_input:
         st.session_state['watsonx_project_id'] = project_id_input
-    
+    if watsonx_url_input:
+        st.session_state['watsonx_url'] = watsonx_url_input
+
     # Main input area
     user_url = st.text_input('Provide a URL')
     # UI component to enter the question
@@ -50,14 +55,14 @@ def main():
     
     collection_name = "base"
     client = utils.chromadb_client()
-    
-    if st.session_state['api_key'] and st.session_state['watsonx_project_id']:
+
+    if st.session_state['api_key'] and st.session_state['watsonx_project_id'] and st.session_state['watsonx_url']:
         if button_clicked and user_url:
             # Invoke the LLM when the button is clicked
-            response = webchat.answer_questions_from_web(st.session_state['api_key'], st.session_state['watsonx_project_id'], user_url, question, collection_name, client)
+            response = webchat.answer_questions_from_web(st.session_state['api_key'], st.session_state['watsonx_project_id'], user_url, question, collection_name, client, st.session_state['watsonx_url'])
             st.write(response)
     else:
-        st.warning("Please provide API Key and Project ID in the sidebar.")
+        st.warning("Please provide API Key, Project ID, and Watsonx URL in the sidebar.")
   
     # Cleaning Vector Database
     st.sidebar.markdown("<hr>", unsafe_allow_html=True)