Update app.py

app.py

@@ -69,26 +69,46 @@ def get_insta_info(df):
     df['Country'] = ''
     df['Date Joined'] = ''
     
-    for index, row in df.iterrows():
-        print(row['Username'])
-        profile_info = scrape_instagram(row['Username'])
+def get_insta_info(df):
+    # Add new columns to the DataFrame
+    df['Bio'] = ''
+    df['Follower Count'] = 0
+    df['Following Count'] = 0
+    df['Bio Links'] = ''
+    df['Full Name'] = ''
+    df['Username'] = ''
+    df['Num Posts'] = 0
+    df['Profile ID'] = ''
+    df['Email'] = ''
+    df['Badge'] = ''
+    df['Category'] = ''
+    df['Phone Number'] = ''
+    df['City Name'] = ''
+    df['Country'] = ''
+    df['Date Joined'] = ''
+    
+    links = df['Links'].values
+    print(links)
+    
+    for i in range(len(links)):
+        profile_info = scrape_instagram(links[i])
         
         if profile_info:  # Only populate if profile_info is not empty
-            df.at[index, 'Bio'] = profile_info['bio']
-            df.at[index, 'Follower Count'] = profile_info['follower_count']
-            df.at[index, 'Following Count'] = profile_info['following_count']
-            df.at[index, 'Bio Links'] = ', '.join(profile_info['bio_links'])
-            df.at[index, 'Full Name'] = profile_info['full_name']
-            df.at[index, 'Username'] = profile_info['username']
-            df.at[index, 'Num Posts'] = profile_info['num_posts']
-            df.at[index, 'Profile ID'] = profile_info['profile_id']
-            df.at[index, 'Email'] = profile_info['email']
-            df.at[index, 'Badge'] = ', '.join(profile_info['badge'])
-            df.at[index, 'Category'] = profile_info['category']
-            df.at[index, 'Phone Number'] = profile_info['phone_number']
-            df.at[index, 'City Name'] = profile_info['city_name']
-            df.at[index, 'Country'] = profile_info['country']
-            df.at[index, 'Date Joined'] = profile_info['date_joined']
+            df.at[i, 'Bio'] = profile_info['bio']
+            df.at[i, 'Follower Count'] = profile_info['follower_count']
+            df.at[i, 'Following Count'] = profile_info['following_count']
+            df.at[i, 'Bio Links'] = ', '.join(profile_info['bio_links'])
+            df.at[i, 'Full Name'] = profile_info['full_name']
+            df.at[i, 'Username'] = profile_info['username']
+            df.at[i, 'Num Posts'] = profile_info['num_posts']
+            df.at[i, 'Profile ID'] = profile_info['profile_id']
+            df.at[i, 'Email'] = profile_info['email']
+            df.at[i, 'Badge'] = ', '.join(profile_info['badge'])
+            df.at[i, 'Category'] = profile_info['category']
+            df.at[i, 'Phone Number'] = profile_info['phone_number']
+            df.at[i, 'City Name'] = profile_info['city_name']
+            df.at[i, 'Country'] = profile_info['country']
+            df.at[i, 'Date Joined'] = profile_info['date_joined']
     
     return df
 
@@ -176,8 +196,8 @@ def get_scrape_data(csv_file, social_media):
     elif social_media == 'Instagram':
         output_df = get_insta_info(df)
     print(output_df.head(2))
-    file_name = './output.csv'
-    output_df.to_csv('./output.csv')
+    file_name = f'./{social_media}_output.csv'
+    output_df.to_csv(f'./{social_media}_output.csv')
     completion_status = "Done"
     return completion_status, gr.DownloadButton(label='Download AI Content', value=file_name, visible=True)