Update app.py

app.py

@@ -154,6 +154,7 @@ def recognize_face(image_path):
         return "Please upload an image."
 
     try:
+        # Assuming load_and_encode function loads and encodes the image
         unknown_encodings = load_and_encode(image_path)
         if not unknown_encodings:
             return "No face found in the provided image."
@@ -165,49 +166,34 @@ def recognize_face(image_path):
                 known_encoding = np.array(data["encoding"])
                 if face_recognition.compare_faces([known_encoding], unknown_encoding)[0]:
                     info = data["info"]
+                    instagram_handle = info.get("instagram_handle")
                     email = info.get("email", "Email not provided")
-                    insta_handle = info["instagram_handle"]
-
-                    # Instagram API request
-                    url = "https://instagram-scraper-api2.p.rapidapi.com/v1/info"
-                    querystring = {"username_or_id_or_url": insta_handle}
-                    headers = {
-                        "x-rapidapi-key": "bc2b15f577msh345ac5a5b422c7ep106f73jsn580b1c25e0d9",
-                        "x-rapidapi-host": "instagram-scraper-api2.p.rapidapi.com"
-                    }
-
-                    try:
-                        response = requests.get(url, headers=headers, params=querystring)
-                        response.raise_for_status()  # Raise an exception for HTTP errors
-                        data = response.json().get('data', {})
-                        print("API Response Data:", data)  # Debug print
-                        full_name = data.get('full_name', 'N/A')
-                        biography = data.get('biography', 'N/A')
-                        follower_count = data.get('follower_count', 'N/A')
-                        following_count = data.get('following_count', 'N/A')
-                    except requests.exceptions.HTTPError as http_err:
-                        print(f"HTTP error occurred: {http_err}")
-                        full_name, biography, follower_count, following_count = 'N/A', 'N/A', 'N/A', 'N/A'
-                    except Exception as err:
-                        print(f"Other error occurred: {err}")
-                        full_name, biography, follower_count, following_count = 'N/A', 'N/A', 'N/A', 'N/A'
-
-                    face_matches.append((name, insta_handle, email, full_name, biography, follower_count, following_count))
+                    
+                    # Fetch additional Instagram data
+                    if instagram_handle:
+                        insta_data = fetch_instagram_data(instagram_handle)
+                        if insta_data:
+                            edge_followed_by = insta_data.get('edge_followed_by', {}).get('count')
+                            edge_follow = insta_data.get('edge_follow', {}).get('count')
+                            full_name = insta_data.get('full_name')
+                            biography = insta_data.get('biography')
+                            face_matches.append((name, instagram_handle, email, edge_followed_by, edge_follow, full_name, biography))
+                        else:
+                            face_matches.append((name, instagram_handle, email, "N/A", "N/A", "Unknown", "Unknown"))
+                    else:
+                        face_matches.append((name, "Unknown", email, "N/A", "N/A", "Unknown", "Unknown"))
 
             if face_matches:
                 matches.extend(face_matches)
             else:
-                matches.append(("Unknown", "Unknown", "Unknown", 'N/A', 'N/A', 'N/A', 'N/A'))
+                matches.append(("Unknown", "Unknown", "Unknown", "N/A", "N/A", "Unknown", "Unknown"))
 
         if matches:
             results = []
-            for name, insta_handle, email, full_name, biography, follower_count, following_count in matches:
+            for name, insta_handle, email, edge_followed_by, edge_follow, full_name, biography in matches:
                 insta_link = f"https://www.instagram.com/{insta_handle}/"
                 insta_link_html = f'<a href="{insta_link}" target="_blank"><font color="red">{insta_handle}</font></a>'
-                results.append(
-                    f"- It's a picture of {name}! Insta handle: {insta_link_html}, Email: {email}, "
-                    f"Full Name: {full_name}, Biography: {biography}, Followers: {follower_count}, Following: {following_count}"
-                )
+                results.append(f"- It's a picture of {name}! Insta handle: {insta_link_html}, Email: {email}, Followers: {edge_followed_by}, Following: {edge_follow}, Full Name: {full_name}, Biography: {biography}")
             log_action(st.session_state.auth_state["user"].email, "Recognized face")
             return "\n".join(results)
         else:
@@ -215,6 +201,27 @@ def recognize_face(image_path):
     except Exception as e:
         return f"Failed to recognize face: {str(e)}"
 
+def fetch_instagram_data(instagram_handle):
+    url = f"https://instagram-scraper-20231.p.rapidapi.com/userinfo/{instagram_handle}"
+    headers = {
+        "x-rapidapi-key": "bc2b15f577msh345ac5a5b422c7ep106f73jsn580b1c25e0d9",
+        "x-rapidapi-host": "instagram-scraper-20231.p.rapidapi.com"
+    }
+
+    try:
+        response = requests.get(url, headers=headers)
+        if response.status_code == 200:
+            data = response.json()
+            if data.get('status') == 'success':
+                return data.get('data', {})
+            else:
+                return None
+        else:
+            return None
+    except Exception as e:
+        print(f"Error fetching Instagram data: {str(e)}")
+        return None
+
         
 # Update recognize_face_optimal function to handle multiple face encodings
 def recognize_face_optimal(image_path):