Spaces:
Sleeping
Sleeping
File size: 2,906 Bytes
43eb61c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
import { useState } from 'react';
import { Image as ImageIcon, Loader2, Sparkles } from 'lucide-react';
export default function ComfyUI() {
const [prompt, setPrompt] = useState('');
const [loading, setLoading] = useState(false);
const [imageUrl, setImageUrl] = useState('');
const [error, setError] = useState('');
const handleGenerate = async () => {
if (!prompt.trim()) return;
setLoading(true);
setError('');
setImageUrl('');
try {
const response = await fetch('/api/comfy', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ prompt }),
});
if (!response.ok) throw new Error('Failed to generate image');
const data = await response.json();
setImageUrl(data.url);
} catch (err) {
setError(err.message || 'Something went wrong');
} finally {
setLoading(false);
}
};
return (
<div className="bg-slate-800 rounded-xl p-6 shadow-xl border border-slate-700">
<div className="flex items-center gap-2 mb-4">
<ImageIcon className="w-5 h-5 text-pink-500" />
<h2 className="text-lg font-semibold text-white">ComfyUI Image Gen</h2>
</div>
<div className="space-y-4">
<div>
<label className="block text-sm font-medium text-slate-400 mb-1">
Prompt
</label>
<textarea
value={prompt}
onChange={(e) => setPrompt(e.target.value)}
placeholder="A futuristic cyberpunk city with neon lights..."
className="w-full bg-slate-900 border border-slate-600 rounded-lg p-3 text-white placeholder-slate-500 focus:ring-2 focus:ring-pink-500 focus:border-transparent outline-none transition-all resize-none h-24"
/>
</div>
<button
onClick={handleGenerate}
disabled={loading}
className="w-full bg-gradient-to-r from-pink-600 to-purple-600 hover:from-pink-500 hover:to-purple-500 text-white font-medium py-2.5 rounded-lg transition-all flex items-center justify-center gap-2 disabled:opacity-50 disabled:cursor-not-allowed"
>
{loading ? (
<>
<Loader2 className="w-4 h-4 animate-spin" />
Generating...
</>
) : (
<>
<Sparkles className="w-4 h-4" />
Generate Image
</>
)}
</button>
{error && (
<div className="bg-red-900/30 text-red-400 p-3 rounded-lg text-sm border border-red-800">
{error}
</div>
)}
{imageUrl && (
<div className="mt-4 border border-slate-600 rounded-lg overflow-hidden bg-slate-900">
<img src={imageUrl} alt="Generated content" className="w-full h-auto object-cover" />
</div>
)}
</div>
</div>
);
} |