awacke1 commited on
Commit
6b6ee18
·
verified ·
1 Parent(s): 8184a3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -14,7 +14,7 @@ import tempfile
14
  import shutil
15
  import os
16
 
17
- # Add a login function to handle HuggingFace authentication
18
  def init_huggingface(token: str):
19
  """Initialize HuggingFace with access token."""
20
  try:
@@ -23,7 +23,8 @@ def init_huggingface(token: str):
23
  except Exception as e:
24
  print(f"Error logging in: {str(e)}")
25
  return False
26
- def format_link(item: Dict, number: int, search_type: str) -> str:
 
27
  """Format a link for display in the UI."""
28
  link = item['link']
29
  readme_link = f"{link}/blob/main/README.md"
@@ -208,8 +209,7 @@ def create_repo_zip(data: List[Dict], search_type: str, token: str) -> tuple[str
208
 
209
  return download_link, f"Successfully downloaded {len(successful_downloads)} repositories"
210
 
211
-
212
-
213
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
214
  gr.Markdown("""
215
  # Search the Hugging Face Hub
@@ -271,12 +271,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
271
  aggregated = SwarmyTime(data)
272
  html_results = display_results(df)
273
  return [
274
- html_results, # results_html
275
- "Status: Ready to download", # download_status
276
- "", # download_area
277
- aggregated, # aggregated_output
278
- search_type, # search_type_state
279
- data # current_results
280
  ]
281
 
282
  async def handle_readme_download(data, token):
 
14
  import shutil
15
  import os
16
 
17
+ # Initialize HuggingFace with access token
18
  def init_huggingface(token: str):
19
  """Initialize HuggingFace with access token."""
20
  try:
 
23
  except Exception as e:
24
  print(f"Error logging in: {str(e)}")
25
  return False
26
+
27
+ def format_link(item: Dict, number: int, search_type: str) -> str:
28
  """Format a link for display in the UI."""
29
  link = item['link']
30
  readme_link = f"{link}/blob/main/README.md"
 
209
 
210
  return download_link, f"Successfully downloaded {len(successful_downloads)} repositories"
211
 
212
+ # Gradio Interface
 
213
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
214
  gr.Markdown("""
215
  # Search the Hugging Face Hub
 
271
  aggregated = SwarmyTime(data)
272
  html_results = display_results(df)
273
  return [
274
+ html_results,
275
+ "Status: Ready to download",
276
+ "",
277
+ aggregated,
278
+ search_type,
279
+ data
280
  ]
281
 
282
  async def handle_readme_download(data, token):