Update app.py

app.py

@@ -108,12 +108,20 @@ def details_fn(query):
         <div class='im_container'>
         """
         images = soup.findAll('img')
+        if query.startswith("https://"):
+            q_beg="https://"
+        if query.startswith("http://"):
+            q_beg="http://"
+        query_src=query.split("//",1)[1]
+        if "/" in query_src:
+            query_src.split("/",1)[0]
+        query_comp=f'{q_beg}{query_src}'
         for i,img in enumerate(images):
             if not img['src'].startswith("data:"):
                 imga=img['src']
                 print(imga)
                 if imga.startswith("/"):
-                    imga = f'{query.strip("/") if query.endswith("/") else query}{imga}'
+                    imga = f'{query_comp}{imga}'
                 im_id = f'img_{i}'
                 link_list.append(imga)
                 image_out += f"""