Update app.py

app.py

@@ -1,7 +1,9 @@
 import streamlit as st
 from huggingface_hub import HfApi
 import pandas as pd
-from concurrent.futures import ThreadPoolExecutor, as_completed
+import asyncio
+
+api = HfApi()
 
 # Default list of Hugging Face usernames
 default_users = {
@@ -15,14 +17,11 @@ default_users = {
     ]
 }
 
-api = HfApi()
-
-def get_user_content(username):
+async def fetch_user_content(username):
     try:
-        # Fetch models, datasets, and spaces associated with the user
-        models = api.list_models(author=username)
-        datasets = api.list_datasets(author=username)
-        spaces = api.list_spaces(author=username)
+        models = await asyncio.to_thread(api.list_models, author=username)
+        datasets = await asyncio.to_thread(api.list_datasets, author=username)
+        spaces = await asyncio.to_thread(api.list_spaces, author=username)
         
         return {
             "username": username,
@@ -33,6 +32,10 @@ def get_user_content(username):
     except Exception as e:
         return {"username": username, "error": str(e)}
 
+async def fetch_all_users(usernames):
+    tasks = [fetch_user_content(username) for username in usernames]
+    return await asyncio.gather(*tasks)
+
 st.title("Hugging Face User Content Display")
 
 # Convert the default users list to a string
@@ -44,24 +47,8 @@ usernames = st.text_area("Enter Hugging Face usernames (one per line):", value=d
 if st.button("Show User Content"):
     if usernames:
         username_list = [username.strip() for username in usernames.split('\n') if username.strip()]
-        results = []
-        status_bars = {}
-
-        # Set up the progress bars for each user
-        for username in username_list:
-            status_bars[username] = st.progress(0, text=f"Fetching data for {username}...")
-
-        def fetch_and_display(username):
-            content = get_user_content(username)
-            status_bars[username].progress(100, text=f"Data fetched for {username}")
-            return content
 
-        # Use ThreadPoolExecutor for concurrent execution
-        with ThreadPoolExecutor(max_workers=len(username_list)) as executor:
-            future_to_username = {executor.submit(fetch_and_display, username): username for username in username_list}
-            for future in as_completed(future_to_username):
-                result = future.result()
-                results.append(result)
+        results = asyncio.run(fetch_all_users(username_list))
 
         st.markdown("### User Content Overview")
         for result in results:
@@ -93,5 +80,4 @@ st.sidebar.markdown("""
 1. The text area is pre-filled with a list of Hugging Face usernames. You can edit this list or add more usernames.
 2. Click 'Show User Content'.
 3. View the user's models, datasets, and spaces along with a link to their Hugging Face profile.
-4. The progress bars show the status of content retrieval for each user.
 """)