discussions

app.py

@@ -1,7 +1,10 @@
 import os
 import secrets
 import urllib.parse
-from flask import Flask, redirect, request, session, render_template, url_for
+import sqlite3
+import json
+import time
+from flask import Flask, redirect, request, session, render_template, url_for, g
 import requests
 
 app = Flask(__name__)
@@ -17,28 +20,158 @@ OAUTH_CLIENT_SECRET = os.environ.get("OAUTH_CLIENT_SECRET")
 OPENID_PROVIDER_URL = os.environ.get("OPENID_PROVIDER_URL", "https://huggingface.co")
 SPACE_HOST = os.environ.get("SPACE_HOST", "localhost:7860")
 
-# Determine the base URL for redirects
+# Cache settings
+CACHE_TTL = 300  # 5 minutes
+DB_PATH = "cache.db"
+
+
+def get_db():
+    if "db" not in g:
+        g.db = sqlite3.connect(DB_PATH)
+        g.db.row_factory = sqlite3.Row
+    return g.db
+
+
+@app.teardown_appcontext
+def close_db(exception):
+    db = g.pop("db", None)
+    if db is not None:
+        db.close()
+
+
+def init_db():
+    with sqlite3.connect(DB_PATH) as conn:
+        conn.execute("""
+            CREATE TABLE IF NOT EXISTS cache (
+                key TEXT PRIMARY KEY,
+                value TEXT,
+                expires_at REAL
+            )
+        """)
+        conn.commit()
+
+
+def cache_get(key):
+    db = get_db()
+    row = db.execute(
+        "SELECT value, expires_at FROM cache WHERE key = ?", (key,)
+    ).fetchone()
+    if row and row["expires_at"] > time.time():
+        return json.loads(row["value"])
+    return None
+
+
+def cache_set(key, value, ttl=CACHE_TTL):
+    db = get_db()
+    expires_at = time.time() + ttl
+    db.execute(
+        "INSERT OR REPLACE INTO cache (key, value, expires_at) VALUES (?, ?, ?)",
+        (key, json.dumps(value), expires_at),
+    )
+    db.commit()
+
+
 def get_base_url():
     if "localhost" in SPACE_HOST or "127.0.0.1" in SPACE_HOST:
         return f"http://{SPACE_HOST}"
     return f"https://{SPACE_HOST}"
 
+
+def fetch_user_spaces(username, token=None):
+    cache_key = f"spaces:{username}"
+    cached = cache_get(cache_key)
+    if cached is not None:
+        return cached
+
+    headers = {}
+    if token:
+        headers["Authorization"] = f"Bearer {token}"
+
+    spaces = []
+    url = f"https://huggingface.co/api/spaces?author={username}"
+
+    try:
+        resp = requests.get(url, headers=headers, timeout=10)
+        if resp.status_code == 200:
+            spaces = resp.json()
+    except Exception:
+        pass
+
+    cache_set(cache_key, spaces)
+    return spaces
+
+
+def fetch_space_discussions(space_id, token=None):
+    cache_key = f"discussions:{space_id}"
+    cached = cache_get(cache_key)
+    if cached is not None:
+        return cached
+
+    headers = {}
+    if token:
+        headers["Authorization"] = f"Bearer {token}"
+
+    discussions = []
+    url = f"https://huggingface.co/api/spaces/{space_id}/discussions"
+
+    try:
+        resp = requests.get(url, headers=headers, timeout=10)
+        if resp.status_code == 200:
+            data = resp.json()
+            discussions = data.get("discussions", [])
+    except Exception:
+        pass
+
+    cache_set(cache_key, discussions)
+    return discussions
+
+
+def get_discussions_feed(username, token=None):
+    spaces = fetch_user_spaces(username, token)
+
+    all_discussions = []
+    for space in spaces:
+        space_id = space.get("id", "")
+        discussions = fetch_space_discussions(space_id, token)
+
+        for d in discussions:
+            score = d.get("numComments", 0) + d.get("numReactionUsers", 0) * 2
+            all_discussions.append({
+                "space_id": space_id,
+                "space_name": space_id.split("/")[-1] if "/" in space_id else space_id,
+                "num": d.get("num"),
+                "title": d.get("title"),
+                "status": d.get("status"),
+                "is_pr": d.get("isPullRequest", False),
+                "author": d.get("author", {}).get("name", "unknown"),
+                "author_avatar": d.get("author", {}).get("avatarUrl", ""),
+                "created_at": d.get("createdAt", ""),
+                "num_comments": d.get("numComments", 0),
+                "num_reactions": d.get("numReactionUsers", 0),
+                "top_reactions": d.get("topReactions", []),
+                "score": score,
+                "url": f"https://huggingface.co/spaces/{space_id}/discussions/{d.get('num')}",
+            })
+
+    all_discussions.sort(key=lambda x: x["score"], reverse=True)
+    return all_discussions
+
+
 @app.route("/")
 def index():
     if "user" in session:
         return redirect(url_for("dashboard"))
     return render_template("index.html")
 
+
 @app.route("/login")
 def login():
     if not OAUTH_CLIENT_ID:
         return "OAuth not configured. Make sure hf_oauth: true is set in your Space's README.md", 500
 
-    # Generate state for CSRF protection
     state = secrets.token_urlsafe(32)
     session["oauth_state"] = state
 
-    # Build authorization URL
     redirect_uri = f"{get_base_url()}/login/callback"
     params = {
         "client_id": OAUTH_CLIENT_ID,
@@ -51,20 +184,18 @@ def login():
 
     return redirect(auth_url)
 
+
 @app.route("/login/callback")
 def callback():
-    # Verify state
     state = request.args.get("state")
     if state != session.get("oauth_state"):
         return "Invalid state parameter", 400
 
-    # Get authorization code
     code = request.args.get("code")
     if not code:
         error = request.args.get("error", "Unknown error")
         return f"Authorization failed: {error}", 400
 
-    # Exchange code for tokens
     redirect_uri = f"{get_base_url()}/login/callback"
     token_url = f"{OPENID_PROVIDER_URL}/oauth/token"
 
@@ -88,7 +219,6 @@ def callback():
     tokens = token_response.json()
     access_token = tokens.get("access_token")
 
-    # Get user info
     userinfo_url = f"{OPENID_PROVIDER_URL}/oauth/userinfo"
     userinfo_response = requests.get(
         userinfo_url,
@@ -100,7 +230,6 @@ def callback():
 
     user_info = userinfo_response.json()
 
-    # Store user info in session
     session["user"] = {
         "sub": user_info.get("sub"),
         "username": user_info.get("preferred_username"),
@@ -110,21 +239,38 @@ def callback():
     }
     session["access_token"] = access_token
 
-    # Clean up state
     session.pop("oauth_state", None)
 
     return redirect(url_for("dashboard"))
 
+
 @app.route("/dashboard")
 def dashboard():
     if "user" not in session:
         return redirect(url_for("index"))
-    return render_template("dashboard.html", user=session["user"])
+
+    username = session["user"]["username"]
+    token = session.get("access_token")
+
+    spaces = fetch_user_spaces(username, token)
+    discussions = get_discussions_feed(username, token)
+
+    return render_template(
+        "dashboard.html",
+        user=session["user"],
+        spaces=spaces,
+        discussions=discussions,
+    )
+
 
 @app.route("/logout")
 def logout():
     session.clear()
     return redirect(url_for("index"))
 
+
+# Initialize database on startup
+init_db()
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)

templates/dashboard.html

@@ -16,12 +16,20 @@
             color: #e5e5e5;
             min-height: 100vh;
         }
+        a {
+            color: inherit;
+            text-decoration: none;
+        }
         .navbar {
             border-bottom: 1px solid #1a1a1a;
             padding: 1rem 2rem;
             display: flex;
             justify-content: space-between;
             align-items: center;
+            position: sticky;
+            top: 0;
+            background: #0a0a0a;
+            z-index: 100;
         }
         .navbar h1 {
             font-size: 0.875rem;
@@ -46,58 +54,148 @@
         .logout-btn {
             color: #666;
             font-size: 0.75rem;
-            text-decoration: none;
             transition: color 0.2s;
         }
         .logout-btn:hover {
             color: #e5e5e5;
         }
         .main {
-            max-width: 600px;
+            max-width: 700px;
             margin: 0 auto;
-            padding: 4rem 2rem;
+            padding: 2rem 1.5rem 4rem;
         }
-        .welcome {
+        .section {
             margin-bottom: 3rem;
         }
-        .welcome h2 {
-            font-size: 1.25rem;
-            font-weight: 500;
-            letter-spacing: -0.02em;
-            margin-bottom: 0.5rem;
+        .section-header {
+            display: flex;
+            justify-content: space-between;
+            align-items: center;
+            margin-bottom: 1rem;
+            padding-bottom: 0.75rem;
+            border-bottom: 1px solid #1a1a1a;
         }
-        .welcome p {
+        .section-title {
+            font-size: 0.75rem;
+            font-weight: 500;
             color: #666;
-            font-size: 0.875rem;
+            text-transform: uppercase;
+            letter-spacing: 0.05em;
+        }
+        .section-count {
+            font-size: 0.75rem;
+            color: #444;
         }
-        .info-list {
+        .spaces-grid {
+            display: flex;
+            flex-wrap: wrap;
+            gap: 0.5rem;
+        }
+        .space-chip {
+            font-size: 0.8125rem;
+            padding: 0.375rem 0.75rem;
+            background: #141414;
+            border: 1px solid #222;
+            border-radius: 4px;
+            color: #999;
+            transition: border-color 0.2s, color 0.2s;
+        }
+        .space-chip:hover {
+            border-color: #333;
+            color: #e5e5e5;
+        }
+        .feed {
             display: flex;
             flex-direction: column;
-            gap: 1px;
-            background: #1a1a1a;
-            border-radius: 8px;
-            overflow: hidden;
         }
-        .info-item {
-            background: #0a0a0a;
-            padding: 1rem 1.25rem;
+        .feed-item {
+            padding: 1rem 0;
+            border-bottom: 1px solid #141414;
+            display: block;
+            transition: background 0.2s;
+        }
+        .feed-item:first-child {
+            padding-top: 0;
+        }
+        .feed-item:last-child {
+            border-bottom: none;
+        }
+        .feed-item:hover {
+            background: #0f0f0f;
+            margin: 0 -1rem;
+            padding-left: 1rem;
+            padding-right: 1rem;
+        }
+        .feed-meta {
             display: flex;
-            justify-content: space-between;
             align-items: center;
+            gap: 0.5rem;
+            margin-bottom: 0.375rem;
         }
-        .info-item:first-child {
-            padding-top: 1.25rem;
+        .feed-space {
+            font-size: 0.75rem;
+            color: #555;
         }
-        .info-item:last-child {
-            padding-bottom: 1.25rem;
+        .feed-type {
+            font-size: 0.625rem;
+            padding: 0.125rem 0.375rem;
+            border-radius: 3px;
+            text-transform: uppercase;
+            letter-spacing: 0.03em;
         }
-        .info-label {
-            color: #666;
-            font-size: 0.8125rem;
+        .feed-type.pr {
+            background: #1a1a2e;
+            color: #6366f1;
         }
-        .info-value {
-            font-size: 0.875rem;
+        .feed-type.discussion {
+            background: #1a2e1a;
+            color: #22c55e;
+        }
+        .feed-status {
+            font-size: 0.625rem;
+            color: #444;
+            text-transform: uppercase;
+        }
+        .feed-status.open { color: #22c55e; }
+        .feed-status.merged { color: #a855f7; }
+        .feed-status.closed { color: #666; }
+        .feed-title {
+            font-size: 0.9375rem;
             color: #e5e5e5;
+            margin-bottom: 0.5rem;
+            line-height: 1.4;
+        }
+        .feed-stats {
+            display: flex;
+            align-items: center;
+            gap: 1rem;
+            font-size: 0.75rem;
+            color: #555;
+        }
+        .feed-stat {
+            display: flex;
+            align-items: center;
+            gap: 0.25rem;
+        }
+        .feed-reactions {
+            display: flex;
+            gap: 0.25rem;
+        }
+        .feed-author {
+            display: flex;
+            align-items: center;
+            gap: 0.375rem;
+        }
+        .feed-author-avatar {
+            width: 16px;
+            height: 16px;
+            border-radius: 50%;
+        }
+        .empty {
+            text-align: center;
+            padding: 3rem 1rem;
+            color: #444;
+            font-size: 0.875rem;
         }
     </style>
 </head>
@@ -114,33 +212,67 @@
     </nav>
 
     <main class="main">
-        <div class="welcome">
-            <h2>{{ user.name or user.username }}</h2>
-            <p>Signed in with Hugging Face</p>
-        </div>
-
-        <div class="info-list">
-            <div class="info-item">
-                <span class="info-label">Username</span>
-                <span class="info-value">{{ user.username }}</span>
+        <section class="section">
+            <div class="section-header">
+                <h2 class="section-title">Your Spaces</h2>
+                <span class="section-count">{{ spaces|length }}</span>
             </div>
-            {% if user.name %}
-            <div class="info-item">
-                <span class="info-label">Name</span>
-                <span class="info-value">{{ user.name }}</span>
+            {% if spaces %}
+            <div class="spaces-grid">
+                {% for space in spaces %}
+                <a href="https://huggingface.co/spaces/{{ space.id }}" target="_blank" class="space-chip">
+                    {{ space.id.split('/')[-1] }}
+                </a>
+                {% endfor %}
             </div>
+            {% else %}
+            <div class="empty">No spaces found</div>
             {% endif %}
-            {% if user.email %}
-            <div class="info-item">
-                <span class="info-label">Email</span>
-                <span class="info-value">{{ user.email }}</span>
+        </section>
+
+        <section class="section">
+            <div class="section-header">
+                <h2 class="section-title">Discussions Feed</h2>
+                <span class="section-count">{{ discussions|length }}</span>
             </div>
-            {% endif %}
-            <div class="info-item">
-                <span class="info-label">ID</span>
-                <span class="info-value">{{ user.sub }}</span>
+            {% if discussions %}
+            <div class="feed">
+                {% for d in discussions %}
+                <a href="{{ d.url }}" target="_blank" class="feed-item">
+                    <div class="feed-meta">
+                        <span class="feed-space">{{ d.space_name }}</span>
+                        <span class="feed-type {{ 'pr' if d.is_pr else 'discussion' }}">
+                            {{ 'PR' if d.is_pr else 'Discussion' }}
+                        </span>
+                        <span class="feed-status {{ d.status }}">{{ d.status }}</span>
+                    </div>
+                    <div class="feed-title">{{ d.title }}</div>
+                    <div class="feed-stats">
+                        <span class="feed-author">
+                            {% if d.author_avatar %}
+                            <img src="{{ d.author_avatar if d.author_avatar.startswith('http') else 'https://huggingface.co' + d.author_avatar }}" alt="" class="feed-author-avatar">
+                            {% endif %}
+                            {{ d.author }}
+                        </span>
+                        <span class="feed-stat">{{ d.num_comments }} comment{{ 's' if d.num_comments != 1 else '' }}</span>
+                        {% if d.num_reactions > 0 %}
+                        <span class="feed-stat">
+                            <span class="feed-reactions">
+                                {% for r in d.top_reactions[:3] %}
+                                {{ r.reaction }}
+                                {% endfor %}
+                            </span>
+                            {{ d.num_reactions }}
+                        </span>
+                        {% endif %}
+                    </div>
+                </a>
+                {% endfor %}
             </div>
-        </div>
+            {% else %}
+            <div class="empty">No discussions yet</div>
+            {% endif %}
+        </section>
     </main>
 </body>
 </html>