commit 1

app.py

@@ -7,9 +7,8 @@ import matplotlib.pyplot as plt
 
 access_token = '24001214633baa923e4caf1fba8da62847aa8df94b3b041d40b54186eabd6a16a81962c9bbbd02f52082a'
 
-
-#x = st.slider('Select a value')
-#st.write(x, 'squared is', x * x)
+# x = st.slider('Select a value')
+# st.write(x, 'squared is', x * x)
 
 
 api_version = '5.131'
@@ -18,7 +17,13 @@ read_id = st.text_input("Type interested vk id or write \"0\" to use default: ")
 if read_id != '0':
     my_id = read_id
     st.write("my_id changed", my_id)
-friend_num = int(st.text_input("Type how many friends you want to show or write \"0\" show all: "))
+friend_num_text = st.text_input("Type how many friends you want to show or write \"0\" show all: ")
+friend_num = 0
+try:
+    friend_num_c = int(friend_num_text)
+except:
+    friend_num_c = 15
+
 st.write("wait about a minute we could do only 3 request per second")
 
 params_for_users_get = {'user_ids': my_id,
@@ -26,7 +31,7 @@ params_for_users_get = {'user_ids': my_id,
                         'v': api_version}
 
 myFirstAndLastName = requests.get('https://api.vk.com/method/users.get', params=params_for_users_get)
-#print(myFirstAndLastName.text)
+# print(myFirstAndLastName.text)
 
 params_for_friends_get = {'user_id': my_id,
                           'order': 'hints',
@@ -40,9 +45,9 @@ if 'error' in r.json() and r.json()['error']['error_code'] == 30:
     exit()
 print(r.text)
 AllMyFriends = r.json()['response']['items']
-#print(AllMyFriends[0]['id'])
-#print(AllMyFriends[0]['is_closed'])
-#print(AllMyFriends)
+# print(AllMyFriends[0]['id'])
+# print(AllMyFriends[0]['is_closed'])
+# print(AllMyFriends)
 AllMyFriendsWithoutClosed = [i for i in AllMyFriends if 'is_closed' in i and i['is_closed'] == False]
 if friend_num == 0:
     friend_num = len(AllMyFriendsWithoutClosed)
@@ -71,13 +76,13 @@ def GetFriendListById(friend_id):
         return [], False
 
 
-MyFriends = AllMyFriendsWithoutClosed[:friend_num-1]
-#MyFriends.append(myFirstAndLastName.json()['response'][0])
+MyFriends = AllMyFriendsWithoutClosed[:friend_num - 1]
+# MyFriends.append(myFirstAndLastName.json()['response'][0])
 MyFriendsSet = set([i['id'] for i in MyFriends])
 MyFriendsDict = dict([(i['id'], i) for i in MyFriends])
 MyFriendsDict[int(my_id)] = myFirstAndLastName.json()['response'][0]
-#print(MyFriends)
-#print(G.nodes())
+# print(MyFriends)
+# print(G.nodes())
 for friend_id in MyFriends:
     friendFriendList, Flag = GetFriendListById(friend_id['id'])
     if not Flag:
@@ -88,17 +93,16 @@ for friend_id in MyFriends:
             G.add_edge(friend_id['id'], friendFriend_id)
 
 G_nodes_list = list(G.nodes)
-#print(G_nodes_list)
+# print(G_nodes_list)
 
 pos = nx.spring_layout(G)
-#print(pos)
+# print(pos)
 
 nx.draw_networkx_nodes(G, pos, node_size=50)
 nx.draw_networkx_edges(G, pos, edge_color='b')
 # plt.show()
 
-#st.pyplot(plt)
-
+# st.pyplot(plt)
 
 
 edge_x = []
@@ -151,12 +155,13 @@ node_trace = go.Scatter(
 
 node_adjacencies = []
 node_text = []
-#print(MyFriendsDict)
+# print(MyFriendsDict)
 # print(G_nodes_list[1])
 # print(MyFriendsDict[G_nodes_list[1]]['first_name'])
 for node, adjacencies in enumerate(G.adjacency()):
     node_adjacencies.append(len(adjacencies[1]))
-    node_text.append(MyFriendsDict[G_nodes_list[node]]['first_name'] + ' ' + MyFriendsDict[G_nodes_list[node]]['last_name'] + ' ' + str(len(adjacencies[1])) + ' edges')
+    node_text.append(MyFriendsDict[G_nodes_list[node]]['first_name'] + ' ' + MyFriendsDict[G_nodes_list[node]][
+        'last_name'] + ' ' + str(len(adjacencies[1])) + ' edges')
     #  print(node)
 
 node_trace.marker.color = node_adjacencies
@@ -164,7 +169,8 @@ node_trace.text = node_text
 
 fig = go.Figure(data=[edge_trace, node_trace],
                 layout=go.Layout(
-                    title='<br>Social graph to ' + myFirstAndLastName.json()['response'][0]['first_name'] + ' ' + myFirstAndLastName.json()['response'][0]['last_name'],
+                    title='<br>Social graph to ' + myFirstAndLastName.json()['response'][0]['first_name'] + ' ' +
+                          myFirstAndLastName.json()['response'][0]['last_name'],
                     titlefont_size=16,
                     showlegend=False,
                     hovermode='closest',