Upload 102 files

ui/src/app/api/auth/hf/callback/route.ts

@@ -44,7 +44,7 @@ export async function GET(request: NextRequest) {
     return htmlResponse(script.trim());
   }
 
-  const redirectUri = `${origin}/api/auth/hf/callback`;
+  const redirectUri = process.env.HF_OAUTH_REDIRECT_URI || process.env.NEXT_PUBLIC_HF_OAUTH_REDIRECT_URI || `${origin}/api/auth/hf/callback`;
 
   try {
     const tokenResponse = await fetch(TOKEN_ENDPOINT, {

ui/src/app/api/auth/hf/login/route.ts

@@ -12,7 +12,7 @@ export async function GET(request: NextRequest) {
 
   const state = randomUUID();
   const origin = request.nextUrl.origin;
-  const redirectUri = `${origin}/api/auth/hf/callback`;
+  const redirectUri = process.env.HF_OAUTH_REDIRECT_URI || process.env.NEXT_PUBLIC_HF_OAUTH_REDIRECT_URI || `${origin}/api/auth/hf/callback`;
 
   const authorizeUrl = new URL(HF_AUTHORIZE_URL);
   authorizeUrl.searchParams.set('response_type', 'code');

ui/src/app/dashboard/page.tsx

@@ -1,6 +1,5 @@
 'use client';
 
-import GpuMonitor from '@/components/GPUMonitor';
 import JobsTable from '@/components/JobsTable';
 import { TopBar, MainContent } from '@/components/layout';
 import Link from 'next/link';
@@ -17,23 +16,55 @@ export default function Dashboard() {
         <div>
           <h1 className="text-lg">Dashboard</h1>
         </div>
-        <div className="flex-1 flex items-center justify-end gap-3 pr-2 text-sm text-gray-400">
+        <div className="flex-1" />
+      </TopBar>
+      <MainContent>
+        <div className="border border-gray-800 rounded-xl bg-gray-900 p-6 flex flex-col gap-4">
+          <div>
+            <h2 className="text-xl font-semibold text-gray-100">
+              {isAuthenticated ? `Welcome back, ${namespace || 'creator'}!` : 'Welcome to Ostris AI Toolkit'}
+            </h2>
+            <p className="text-sm text-gray-400 mt-2">
+              {isAuthenticated
+                ? 'You are signed in with Hugging Face and can manage jobs, datasets, and submissions.'
+                : 'Authenticate with Hugging Face or add a personal access token to create jobs, upload datasets, and launch training.'}
+            </p>
+          </div>
           {isAuthenticated ? (
-            <span>Welcome, {namespace || 'user'}</span>
-          ) : (
-            <>
-              <span>Welcome, Guest</span>
-              <HFLoginButton size="sm" />
-              <Link href="/settings" className="text-xs text-blue-400 hover:text-blue-300">
+            <div className="flex flex-wrap items-center gap-3 text-sm">
+              <Link
+                href="/jobs/new"
+                className="px-4 py-2 rounded-md bg-blue-600 hover:bg-blue-500 text-white transition-colors"
+              >
+                Create a Training Job
+              </Link>
+              <Link
+                href="/datasets"
+                className="px-4 py-2 rounded-md bg-gray-800 hover:bg-gray-700 text-gray-200 transition-colors"
+              >
+                Manage Datasets
+              </Link>
+              <Link
+                href="/settings"
+                className="px-4 py-2 rounded-md border border-gray-700 text-gray-300 hover:border-gray-600 transition-colors"
+              >
                 Settings
               </Link>
-            </>
+            </div>
+          ) : (
+            <div className="flex flex-wrap items-center gap-3 text-sm">
+              <HFLoginButton size="md" />
+              <Link
+                href="/settings"
+                className="text-xs text-blue-400 hover:text-blue-300"
+              >
+                Or manage tokens in Settings
+              </Link>
+            </div>
           )}
         </div>
-      </TopBar>
-      <MainContent>
-        <GpuMonitor />
-        <div className="w-full mt-4">
+
+        <div className="w-full mt-6">
           <div className="flex justify-between items-center mb-2">
             <h1 className="text-md">Active Jobs</h1>
             <div className="text-xs text-gray-500">

ui/src/utils/env.ts

@@ -7,3 +7,4 @@ export const usingBrowserDb = DB_MODE === 'browser';
 export const usingServerDb = DB_MODE === 'server';
 
 export const oauthClientId = process.env.NEXT_PUBLIC_HF_OAUTH_CLIENT_ID || '';
+export const oauthRedirectUri = process.env.NEXT_PUBLIC_HF_OAUTH_REDIRECT_URI || '';