Skip to content

Commit

Permalink
improve design
Browse files Browse the repository at this point in the history
  • Loading branch information
thewh1teagle committed Jun 22, 2024
1 parent ce86e67 commit fda51db
Showing 1 changed file with 36 additions and 35 deletions.
71 changes: 36 additions & 35 deletions src/pages/Setup.tsx
Original file line number Diff line number Diff line change
@@ -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>
)
}

0 comments on commit fda51db

Please sign in to comment.