auto url counter

src/modules/app.py

@@ -1,6 +1,7 @@
-import gradio as gr
-import pandas as pd
+import re
 import itertools
+import pandas as pd
+import gradio as gr
 
 from link import UrlProcessor
 # from main import TextProcessor
@@ -31,7 +32,6 @@ def calculate_positions(image_count, item_size=(20, 20), page_size=(91, 55), pad
     print(positions)
     return positions
 
-
 with gr.Blocks() as demo:
     gr.Markdown(description)
     with gr.Row():
@@ -39,6 +39,11 @@ with gr.Blocks() as demo:
             access_token = gr.Textbox(label="Access Token", placeholder="***")
             api_url = gr.Textbox(label="API base URL", placeholder="https://")
             txt_input = gr.TextArea(label="Enter Text with URLs")
+            count_url = gr.Dropdown(visible=False)
+            @gr.render(inputs=[txt_input])
+            def update_cnt(txt):
+                urls = re.findall(r'http[s]?://\S+', txt)
+                return urls
 
             with gr.Row():
                 shorten_checkbox = gr.Checkbox(label="Shorten URL", value=True)
@@ -51,16 +56,20 @@ with gr.Blocks() as demo:
                     idv_img_input = gr.Files(label='Insert individual images')
 
                 headers = ['ImageId', 'PosX', 'PosY', 'PosPx', 'PosPy']
-                @gr.render(inputs=[cm_img_input, idv_img_input])
-                def update_df(*files):
-                    image_count = sum(len(file) for file in files if file is not None)
-                    positions = calculate_positions(image_count)
+
+                @gr.render(inputs=[count_url, cm_img_input, idv_img_input])
+                def update_df(*items):
+                    image_count = sum(len(item) for item in items if item is not None)
+                    if image_count is not None:
+                        positions = calculate_positions(image_count)
                     df = pd.DataFrame(positions, columns=['PosX', 'PosY', 'PosPx', 'PosPy'])
                     df.insert(0, 'ImageId', range(1, image_count + 1))
                     return df
 
                 pos_df = gr.Dataframe(headers=headers, label="Positions")
 
+                txt_input.change(update_cnt, txt_input, count_url)
+                count_url.change(update_df, count_url, pos_df)
                 cm_img_input.change(update_df, cm_img_input, pos_df)
                 idv_img_input.change(update_df, idv_img_input, pos_df)
 

src/modules/link.py

@@ -100,6 +100,8 @@ class UrlProcessor:
         index_path = 'index.html'
         css_path = 'style.css'
 
+        if urls is not None:
+            urls = urls if urls is not None else []
         for idx, url in enumerate(urls):
             shorten_url, shorten_response_json = None, None
             try:
@@ -161,9 +163,6 @@ class UrlProcessor:
 
         processor = ImageProcessor()
         layer = processor.create_layer()
-        # for qr in qr_svgs:
-        #     with open(qr, "rb") as image_file:
-        #         encoded_string = base64.b64encode(image_file.read()).decode()
 
         os.makedirs(f'{tmpd}/img/cards', exist_ok=True)