Update influencer_ui.py

influencer_ui.py

@@ -4,57 +4,48 @@ import requests
 from io import BytesIO
 import json
 
-
 profile_default_image_url = "https://t4.ftcdn.net/jpg/03/46/93/61/360_F_346936114_RaxE6OQogebgAWTalE1myseY1Hbb5qPM.jpg"
 insta_rag_url = "https://us-central1-steam-cache-277314.cloudfunctions.net/gemini-hackathon-insta-influencer"
-# influencers_data = {
-#     "influencer1": {
-#         "name": "Influencer One",
-#         "image": "./images/inf1.png",  # Replace with actual image paths
-#         "details": "Details about Influencer One..."
-#     },
-#     "influencer2": {
-#         "name": "Influencer Two",
-#         "image": "./images/inf2.png",
-#         "details": "Details about Influencer Two..."
-#     }
-#     # Add more influencers as needed
-# }
 
-            
 def show_influencer_search_page():
     st.header("Influencer Search and Analysis")
     st.markdown("Discover influencers by name, niche, location, follower count, and more.")
     search_query = st.text_input("Search Influencers", "")
     
-    
-    # Influencer search button (assuming this is part of your UI)
+    # Influencer search button
     search_button = st.button("Search Influencers")
     
     if search_button:
         data = {"search_query": search_query}
         resp = requests.post(insta_rag_url, json=data)
-        response_dict = json.loads(resp.text)
-        st.write(response_dict)
-        # 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(img_response.content)
-        influencers_data = {response_dict.get('handle', "No Name"): {"name": response_dict.get('name', 'No Name'),
-                            "image": image_url,
-                            "followers": response_dict.get('followersCount', ''),                         
-                            "details": response_dict.get('bio', '')}}
+        influencers_list = json.loads(resp.text)
 
-        # Display influencer images as clickable buttons
-        for influencer_id, influencer_info in influencers_data.items():
+        for influencer_data in influencers_list:
+            # Download image
+            profile_img_url = influencer_data.get('profile_image_url', profile_default_image_url)
+            img_response = requests.get(profile_img_url)
+            image_url = profile_default_image_url
+            if img_response.status_code == 200:
+                image_url = f"downloaded_image_{influencer_data.get('handle', 'NoName')}.jpg"
+                with open(image_url, "wb") as file:
+                    file.write(img_response.content)
+
+            # Display influencer data
             col1, col2 = st.columns([1, 3])
             with col1:
-                st.image(influencer_info["image"], width=50)
-    
+                st.image(image_url, width=50)
+            
             with col2:
-                with st.expander("Influencer Details"):
-                    st.subheader(influencer_info["name"])
-                    st.write(influencer_info["details"])
+                with st.expander(f"{influencer_data.get('name', 'No Name')}'s Details"):
+                    st.markdown(f"**Handle:** {influencer_data.get('handle', 'No Handle')}")
+                    st.markdown(f"**Bio:** {influencer_data.get('bio', 'No Bio')}")
+                    st.markdown(f"**Email:** {influencer_data.get('email', 'N/A')}")
+                    st.markdown(f"**Country:** {influencer_data.get('country', 'N/A')}")
+                    st.markdown(f"**Location:** {influencer_data.get('location', 'N/A')}")
+                    st.markdown(f"**Followers Count:** {influencer_data.get('followers_count', 'N/A')}")
+                    st.markdown(f"**Engagement Rate:** {influencer_data.get('engagement_rate', 'N/A')*100:.2f}%")
+                    st.markdown("### Frequent Hashtags")
+                    hashtags = influencer_data.get('frequent_hashtags', [])
+                    st.write(", ".join(hashtags))
+                    st.markdown("### Post Summary")
+                    st.write(influencer_data.get('post_summary', 'N/A'))