Update app.py

app.py

@@ -33,7 +33,20 @@ def search_amazon(query: str, num: int = 10) -> str:
 
     return links
 
+import re
 
+def append_tag_to_amazon_url(input_string):
+    pattern = r'(https?://www\.amazon\.com[^)]*)'
+
+    def append_tag(match):
+        url = match.group(0)
+        if '?' in url:
+            return url + '&tag=dpang-20'
+        else:
+            return url + '?tag=dpang-20'
+
+    return re.sub(pattern, append_tag, input_string)
+    
 # Streamlit interface
 st.title('Vision Shop - shopping via images')
 url = st.text_input('Enter the url link to an image', 'Image URL')
@@ -66,8 +79,15 @@ if st.button('Shop'):
         st.write(search_str)
 
         amazon_output = search_amazon(search_str)
-        print(amazon_output)
-        st.write(amazon_output)
+
+        processed_lines = [append_tag_to_amazon_url(line) for line in amazon_output.split('\n')]
+        output_str = '\n'.join(processed_lines)
+
+        print(output_str)
+        st.write(output_str)
+        
+        #print(amazon_output)
+        #st.write(amazon_output)
 
     except Exception as e:
         st.error(f'An error occurred: {e}')