use local

src/modules/bitly-api.min.py

@@ -21,16 +21,18 @@ For more information, see
 
 
 BITLY_ACCESS_TOKEN = os.getenv('BITLY_ACCESS_TOKEN', None)
+BITLY_API_URL = "https://api-ssl.bitly.com/v4"
 
 class ShortenUrl:
-    def __init__(self, bitly_access_token=None):
-        self.bitly_access_token = bitly_access_token if bitly_access_token else BITLY_ACCESS_TOKEN
+    def __init__(self, access_token=None, api_url=None):
+        self.access_token = access_token if access_token else BITLY_ACCESS_TOKEN
+        self.api_url = api_url if api_url else BITLY_API_URL
 
     def shorten(self, long_url):
         try:
-            url = "https://api-ssl.bitly.com/v4/shorten"
+            url = self.api_url + "/shorten"
             headers = {
-                "Authorization": f"Bearer {self.bitly_access_token}",
+                "Authorization": f"Bearer {self.access_token}",
                 "Content-Type": "application/json"
             }
             data = {
@@ -42,16 +44,15 @@ class ShortenUrl:
         except requests.exceptions.RequestException as e:
             return None, {"error": str(e)}
 
-    def generate_qr(self, shorten_url):
+    def generate_qr(self, shorten_url): # DEPRECATED
         try:
-            qr_url = f"https://api-ssl.bitly.com/v4/bitlinks/{shorten_url}/qr"
+            qr_url = self.api_url + f"/bitlinks/{shorten_url}/qr"
             headers = {
-                "Authorization": f"Bearer {self.bitly_access_token}",
-                "Content-Type": "application/json",
+                "Authorization": f"Bearer {self.access_token}",
             }
             params = {
                 "customization": {
-                    "exclude_bitly_logo": True
+                    "exclude_bitly_logo": True # TODO
                 }
             }
             response = requests.get(qr_url, headers=headers, params=params)
@@ -87,8 +88,8 @@ class ShortenUrl:
         except Exception as e:
             return None, None, {"error": str(e)}
 
-def shorten_and_generate_qr(api_key, text, shorten, generate_qr):
-    bitly = ShortenUrl(api_key)
+def shorten_and_generate_qr(access_token, api_url, text, shorten, generate_qr):
+    bitly = ShortenUrl(access_token, api_url)
     urls = re.findall(r'http[s]?://\S+', text)
     results = []
     ts, tmpd = Interface.get_tempdir()
@@ -114,7 +115,8 @@ def shorten_and_generate_qr(api_key, text, shorten, generate_qr):
 
             if generate_qr and url_to_process:
                 if shorten_url:
-                    qr_html, qr_png, qr_json = bitly.generate_qr(url_to_process)
+                    # qr_html, qr_png, qr_json = bitly.generate_qr(url_to_process)
+                    qr_html, qr_png, qr_json = bitly.generate_qr_local(url_to_process)
                 else:
                     qr_html, qr_png, qr_json = bitly.generate_qr_local(url_to_process)
 
@@ -215,7 +217,8 @@ with gr.Blocks() as demo:
     gr.Markdown(description)
     with gr.Row():
         with gr.Column():
-            api_key = gr.Textbox(label="Bitly API Key", placeholder="API Key")
+            access_token = gr.Textbox(label="Access Token", placeholder="***")
+            api_url = gr.Textbox(label="API base URL", placeholder="https://")
             url_input = gr.TextArea(label="Enter Text with URLs")
             with gr.Row():
                 shorten_checkbox = gr.Checkbox(label="Shorten URL", value=True)
@@ -238,7 +241,7 @@ with gr.Blocks() as demo:
 
     submit_button.click(
         shorten_and_generate_qr,
-        inputs=[api_key, url_input, shorten_checkbox, qr_checkbox],
+        inputs=[access_token, api_url, url_input, shorten_checkbox, qr_checkbox],
         outputs=[preview_output, gallery_output, file_output, json_output]
     )