Update app.py

app.py

@@ -2,6 +2,13 @@ import streamlit as st
 import requests
 from urllib.parse import urlencode, quote_plus, urlparse
 
+def is_valid_url(url):
+    """
+    Checks if the URL has a valid scheme.
+    """
+    parsed_url = urlparse(url)
+    return parsed_url.scheme in ['http', 'https']
+
 def forward_parameters(target_url, extra_params):
     """
     Forward parameters to the target URL and return the response.
@@ -10,12 +17,10 @@ def forward_parameters(target_url, extra_params):
     :param extra_params: A dictionary of additional parameters to forward.
     :return: The response from the target URL.
     """
+    if not is_valid_url(target_url):
+        return "Invalid URL: No scheme supplied. URLs must start with http:// or https://"
+    
     try:
-        # Ensure the target URL has a valid scheme
-        parsed_url = urlparse(target_url)
-        if not parsed_url.scheme:
-            return "Invalid URL: No scheme supplied. URLs must start with http:// or https://"
-
         # Construct the full URL with extra parameters
         full_url = f"{target_url}?{urlencode(extra_params, quote_via=quote_plus)}"
         response = requests.get(full_url)  # Using GET request for simplicity
@@ -28,10 +33,9 @@ def forward_parameters(target_url, extra_params):
 st.title('HTTPS Streamlit Parameters Forwarder')
 
 # Correctly accessing the query parameters
-params = st.query_params  # Access query parameters
-st.write("Received parameters:", params)  # Debug: Log received parameters
+params = st.query_params
 
-target_url = params.get('param1', [None])[0]  # Extract the target URL
+target_url = params.get('param1', [None])[0]
 
 if target_url:
     # Remove 'param1' to forward the remaining parameters