|
|
|
import { useState } from "react"; |
|
import { toast } from "sonner"; |
|
import { MdSave } from "react-icons/md"; |
|
import { useParams } from "next/navigation"; |
|
|
|
import Loading from "@/components/loading"; |
|
import { Button } from "@/components/ui/button"; |
|
import { api } from "@/lib/api"; |
|
|
|
export function SaveButton({ |
|
html, |
|
prompts, |
|
}: { |
|
html: string; |
|
prompts: string[]; |
|
}) { |
|
|
|
const { namespace, repoId } = useParams<{ |
|
namespace: string; |
|
repoId: string; |
|
}>(); |
|
const [loading, setLoading] = useState(false); |
|
|
|
const updateSpace = async () => { |
|
setLoading(true); |
|
|
|
try { |
|
const res = await api.put(`/me/projects/${namespace}/${repoId}`, { |
|
html, |
|
prompts, |
|
}); |
|
if (res.data.ok) { |
|
toast.success("Your space is updated! 🎉", { |
|
action: { |
|
label: "See Space", |
|
onClick: () => { |
|
window.open( |
|
`https://huggingface.co/spaces/${namespace}/${repoId}`, |
|
"_blank" |
|
); |
|
}, |
|
}, |
|
}); |
|
} else { |
|
toast.error(res?.data?.error || "Failed to update space"); |
|
} |
|
} catch (err: any) { |
|
toast.error(err.response?.data?.error || err.message); |
|
} finally { |
|
setLoading(false); |
|
} |
|
}; |
|
return ( |
|
<> |
|
<Button |
|
variant="default" |
|
className="max-lg:hidden !px-4 relative" |
|
onClick={updateSpace} |
|
> |
|
<MdSave className="size-4" /> |
|
Deploy your Project{" "} |
|
{loading && <Loading className="ml-2 size-4 animate-spin" />} |
|
</Button> |
|
<Button |
|
variant="default" |
|
size="sm" |
|
className="lg:hidden relative" |
|
onClick={updateSpace} |
|
> |
|
Deploy {loading && <Loading className="ml-2 size-4 animate-spin" />} |
|
</Button> |
|
</> |
|
); |
|
} |
|
|