From fda51dbecf2ee22dab794b52ff0c6dea8ff8fce9 Mon Sep 17 00:00:00 2001
From: thewh1teagle <61390950+thewh1teagle@users.noreply.github.com>
Date: Sat, 22 Jun 2024 16:31:20 +0300
Subject: [PATCH] improve design

---
 src/pages/Setup.tsx | 71 +++++++++++++++++++++++----------------------
 1 file changed, 36 insertions(+), 35 deletions(-)

diff --git a/src/pages/Setup.tsx b/src/pages/Setup.tsx
index ea867ce..14e26e5 100644
--- a/src/pages/Setup.tsx
+++ b/src/pages/Setup.tsx
@@ -1,42 +1,43 @@
-import { invoke } from "@tauri-apps/api/core";
-import { useEffect, useState } from "react";
-import { useNavigate } from "react-router-dom";
-import * as config from "../config";
-import { listen } from "@tauri-apps/api/event";
+import { invoke } from '@tauri-apps/api/core'
+import { useEffect, useState } from 'react'
+import { useNavigate } from 'react-router-dom'
+import * as config from '../config'
+import { listen } from '@tauri-apps/api/event'
 
 export default function Setup() {
-  const [progress, setProgress] = useState<number | null>(null);
-  const navigate = useNavigate();
+	const [progress, setProgress] = useState<number | null>(0)
+	const navigate = useNavigate()
 
-  async function downloadModel() {
-    await new Promise((resolve) => setTimeout(resolve, 1000));
-    await invoke<string>("download_model", {
-      filename: config.modelFilename,
-      url: config.modelURL,
-    });
-    navigate("/");
-  }
+	async function downloadModel() {
+		await new Promise((resolve) => setTimeout(resolve, 1000))
+		await invoke<string>('download_model', {
+			filename: config.modelFilename,
+			url: config.modelURL,
+		})
+		navigate('/')
+	}
 
-  async function listenForProgress() {
-    await listen<[number, number]>("download_progress", (event) => {
-      const [part, total] = event.payload;
-      setProgress((part / total) * 100);
-    });
-  }
+	async function listenForProgress() {
+		await listen<[number, number]>('download_progress', (event) => {
+			const [part, total] = event.payload
+			setProgress((part / total) * 100)
+		})
+	}
 
-  useEffect(() => {
-    listenForProgress();
-    downloadModel();
-  }, []);
+	useEffect(() => {
+		listenForProgress()
+		downloadModel()
+	}, [])
 
-  return (
-    <div>
-      <h1>Downloading model...</h1>
-      <progress
-        className="progress progress-primary w-56"
-        value={progress ?? 0}
-        max="100"
-      ></progress>
-    </div>
-  );
+	return (
+		<div className="w-[100vw] h-[100vh] flex flex-col justify-center items-center">
+			<div className="text-3xl m-5 font-bold">Downloading Meta Model...</div>
+			{progress && progress > 0 && (
+				<>
+					<progress className="progress progress-primary w-56 my-2" value={progress ?? 0} max="100"></progress>
+					<p>This happens only once! 🎉</p>
+				</>
+			)}
+		</div>
+	)
 }