JustAnotherCibrarian commited on
Commit
61c331e
·
verified ·
1 Parent(s): 29f4f30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -108,36 +108,37 @@ with gr.Blocks(title="HF Repo Access Manager") as demo:
108
  # ---Actions---
109
 
110
  # Refresh external data.
111
- def refresh_all_data(token, force = False):
112
  try:
113
  if not token:
114
  raise ValueError("Please login first!")
115
 
116
  # Refresh repos
117
- lrepos = get_user_repos(profile.username, oauth_token.token)
118
 
119
  return gr.Dropdown(choices=lrepos, value=None), gr.Dropdown(choices=lrepos, value=[])
120
  except Exception as e:
121
  gr.Warning(f"Failed to refresh data: {str(e)}")
122
  return gr.Dropdown(choices=[DUMMY_REFRESH], value=None), gr.Dropdown(choices=[DUMMY_REFRESH], value=[])
123
 
124
- def force_refresh(token):
125
  """Run refresh with force.
126
  Lambda method doesn't seem to work for some reason,
127
  perhaps profile and oauth aren't functional with args/kwargs.
128
  """
129
- return refresh_all_data(token, force = True)
130
 
131
  refresh_repos_btn.click(
132
  fn=force_refresh,
133
- inputs=[],
134
  outputs=[repo_input, multi_repo_input]
135
  )
136
 
137
- def check_dummy(token, dropdown_value = None):
138
  """If dummy was selected, initiate dropdown refresh.
139
 
140
  """
 
141
  name = ""
142
  if not dropdown_value: # Empty list or null.
143
  pass
 
108
  # ---Actions---
109
 
110
  # Refresh external data.
111
+ def refresh_all_data(username, token, force = False):
112
  try:
113
  if not token:
114
  raise ValueError("Please login first!")
115
 
116
  # Refresh repos
117
+ lrepos = get_user_repos(username, token)
118
 
119
  return gr.Dropdown(choices=lrepos, value=None), gr.Dropdown(choices=lrepos, value=[])
120
  except Exception as e:
121
  gr.Warning(f"Failed to refresh data: {str(e)}")
122
  return gr.Dropdown(choices=[DUMMY_REFRESH], value=None), gr.Dropdown(choices=[DUMMY_REFRESH], value=[])
123
 
124
+ def force_refresh(username, token):
125
  """Run refresh with force.
126
  Lambda method doesn't seem to work for some reason,
127
  perhaps profile and oauth aren't functional with args/kwargs.
128
  """
129
+ return refresh_all_data(username, token, force = True)
130
 
131
  refresh_repos_btn.click(
132
  fn=force_refresh,
133
+ inputs=[hf_token],
134
  outputs=[repo_input, multi_repo_input]
135
  )
136
 
137
+ def check_dummy(token, dropdown_value = None, request: gr.Request=None):
138
  """If dummy was selected, initiate dropdown refresh.
139
 
140
  """
141
+ gr.Info(request.username if request else "Nonsense")
142
  name = ""
143
  if not dropdown_value: # Empty list or null.
144
  pass