Update influencer_ui.py

influencer_ui.py

@@ -35,10 +35,18 @@ def show_influencer_search_page():
         data = {"search_query": search_query}
         resp = requests.post(insta_rag_url, json=data)
         response_dict = json.loads(resp.text)
+
+        # Download images
+        profile_img_url = response_dict.get('profile_image_url', profile_default_image_url)
+        img_response = requests.get(profile_img_url)
+        if img_response.status_code == 200:
+            image_url = f"downloaded_image_{response_dict.get('handle', "NoName")}.jpg"
+            with open(image_url, "wb") as file:
+                file.write(response.content)
         influencers_data = {response_dict.get('handle', "No Name"): {"name": response_dict.get('name', 'No Name'),
-                            "image": response_dict.get('profile_image_url', profile_default_image_url),
+                            "image": image_url,
                             "details": response_dict.get('bio', '')}}
-        st.write(influencers_data)
+
         # Display influencer images as clickable buttons
         for influencer_id, influencer_info in influencers_data.items():
             col1, col2 = st.columns([1, 3])