Update templates/index.html

templates/index.html

@@ -6,20 +6,23 @@
     <title>استودیو تصویر هوش مصنوعی</title>
     <link href="https://fonts.googleapis.com/css2?family=Vazirmatn:wght@400;600;800&display=swap" rel="stylesheet">
     <style>
-        body { font-family: 'Vazirmatn', Tahoma, sans-serif; background-color: #EDF2F7; padding: 40px 20px; display: flex; justify-content: center; }
-        .container { background-color: #ffffff; border-radius: 12px; box-shadow: 0 10px 15px rgba(0,0,0,0.1); width: 100%; max-width: 500px; padding: 30px; }
+        body { font-family: 'Vazirmatn', Tahoma, sans-serif; background-color: #EDF2F7; padding: 40px 20px; display: flex; justify-content: center; margin: 0;}
+        .container { background-color: #ffffff; border-radius: 12px; box-shadow: 0 10px 15px rgba(0,0,0,0.1); width: 100%; max-width: 600px; padding: 30px; }
         h2 { text-align: center; color: #5A67D8; margin-top: 0; }
         .form-group { margin-bottom: 20px; }
-        label { display: block; margin-bottom: 8px; font-weight: 600; }
-        textarea { width: 100%; padding: 12px; border: 1px solid #E2E8F0; border-radius: 8px; box-sizing: border-box; background: #F7FAFC; }
+        label { display: block; margin-bottom: 8px; font-weight: 600; font-size: 0.95rem;}
+        textarea, select { width: 100%; padding: 12px; border: 1px solid #E2E8F0; border-radius: 8px; box-sizing: border-box; background: #F7FAFC; font-family: inherit;}
         .size-grid { display: grid; grid-template-columns: repeat(2, 1fr); gap: 10px; }
-        .size-btn { padding: 10px; border: 1px solid #E2E8F0; border-radius: 8px; background: #F7FAFC; cursor: pointer; font-weight: 600; transition: 0.2s; }
+        .size-btn { padding: 10px; border: 1px solid #E2E8F0; border-radius: 8px; background: #F7FAFC; cursor: pointer; font-weight: 600; transition: 0.2s; font-family: inherit;}
         .size-btn.active { background: #5A67D8; color: white; border-color: #5A67D8; }
         button.main-btn { width: 100%; padding: 14px; background: #5A67D8; color: white; border: none; border-radius: 8px; font-size: 1.1rem; font-weight: bold; cursor: pointer; }
         button.main-btn:disabled { background: #A3B1E1; cursor: not-allowed; }
         #status-box { margin-top: 20px; padding: 15px; border-radius: 8px; display: none; text-align: center; background: #EBF4FF; color: #3182CE; font-weight: bold; }
-        #result-container { margin-top: 20px; display: none; text-align: center; }
-        #result-container img { max-width: 100%; border-radius: 8px; border: 2px solid #E2E8F0; }
+        
+        .result-grid { display: grid; grid-template-columns: repeat(2, 1fr); gap: 15px; margin-top: 20px; }
+        .result-item { width: 100%; border-radius: 8px; overflow: hidden; border: 2px solid #E2E8F0; box-shadow: 0 4px 6px rgba(0,0,0,0.1); animation: fadeIn 0.5s ease; }
+        .result-item img { width: 100%; height: auto; display: block; }
+        @keyframes fadeIn { from { opacity: 0; transform: scale(0.9); } to { opacity: 1; transform: scale(1); } }
     </style>
 </head>
 <body>
@@ -27,25 +30,39 @@
     <h2>استودیو تصویر هوش مصنوعی</h2>
     <div class="form-group">
         <label>دستور ساخت تصویر (Prompt)</label>
-        <textarea id="prompt" rows="4" placeholder="مثال: A futuristic cyberpunk city..."></textarea>
+        <textarea id="prompt" rows="3" placeholder="مثال: A futuristic cyberpunk city..."></textarea>
+    </div>
+    
+    <div style="display: flex; gap: 15px; margin-bottom: 20px;">
+        <div style="flex: 1;">
+            <label>تعداد تصاویر</label>
+            <select id="num-images">
+                <option value="1">۱ تصویر (سریع)</option>
+                <option value="2">۲ تصویر</option>
+                <option value="3">۳ تصویر</option>
+                <option value="4" selected>۴ تصویر (پیش‌فرض)</option>
+            </select>
+        </div>
     </div>
+
     <div class="form-group">
         <label>ابعاد تصویر</label>
         <div class="size-grid">
-            <button class="size-btn active" data-w="1024" data-h="1024" onclick="selectSize(this)">۱:۱ مربع</button>
-            <button class="size-btn" data-w="1344" data-h="768" onclick="selectSize(this)">۱۶:۹ افقی</button>
+            <button class="size-btn active" data-w="1024" data-h="1024" onclick="selectSize(this)">۱:۱ (مربع)</button>
+            <button class="size-btn" data-w="1344" data-h="768" onclick="selectSize(this)">۱۶:۹ (افقی)</button>
+            <button class="size-btn" data-w="768" data-h="1344" onclick="selectSize(this)">۹:۱۶ (عمودی)</button>
+            <button class="size-btn" data-w="1152" data-h="864" onclick="selectSize(this)">۴:۳ (کلاسیک)</button>
         </div>
     </div>
-    <button id="btn-submit" class="main-btn" onclick="generateImage()">🚀 ساخت تصویر</button>
+    
+    <button id="btn-submit" class="main-btn" onclick="generateImage()">🚀 ساخت تصاویر</button>
+    
     <div id="status-box"></div>
-    <div id="result-container">
-        <h3 style="color: #2ea44f;">🎉 تصویر شما آماده است!</h3>
-        <img id="generated-image" src="" alt="AI Image">
-    </div>
+    <div id="result-container" class="result-grid"></div>
 </div>
 
 <script>
-    let w = 1024, h = 1024, timer;
+    let w = 1024, h = 1024, timer, expectedTotal = 1, receivedImages = 0;
 
     function selectSize(btn) {
         document.querySelectorAll('.size-btn').forEach(b => b.classList.remove('active'));
@@ -56,25 +73,28 @@
 
     async function generateImage() {
         const prompt = document.getElementById('prompt').value.trim();
+        const numImgs = parseInt(document.getElementById('num-images').value);
         if (!prompt) return alert('پرامپت خالی است!');
         
+        expectedTotal = numImgs;
+        receivedImages = 0;
+
         document.getElementById('btn-submit').disabled = true;
-        document.getElementById('result-container').style.display = 'none';
+        document.getElementById('result-container').innerHTML = ''; // پاک کردن نتایج قبلی
+        
         const statusBox = document.getElementById('status-box');
         statusBox.style.display = 'block';
-        statusBox.innerHTML = 'در حال ارتباط با سرور... ⏳';
+        statusBox.innerHTML = 'در حال راه‌اندازی سرور پردازش... ⏳';
 
         try {
-            // درخواست به بک‌اندِ داکر اسپیسِ خودمان
             const res = await fetch('/api/generate', {
                 method: 'POST', headers: {'Content-Type': 'application/json'},
-                body: JSON.stringify({ prompt, width: w, height: h })
+                body: JSON.stringify({ prompt, width: w, height: h, num_images: numImgs })
             });
             const data = await res.json();
             
             if (data.status === 'success') {
-                statusBox.innerHTML = 'گیت‌هاب در حال ساخت و آپلود تصویر است... (حدود ۳۰ تا ۶۰ ثانیه) 🎨';
-                // بررسی دوره‌ای تا زمانی که گیت هاب عکس را در سرور آپلود کند
+                statusBox.innerHTML = `در حال ساخت ${numImgs} تصویر... لطفاً صفحه را نبندید 🎨`;
                 timer = setInterval(() => checkStatus(data.run_id), 3000);
             } else {
                 statusBox.innerHTML = 'خطا در ارسال به گیت‌هاب!';
@@ -87,14 +107,30 @@
     }
 
     async function checkStatus(run_id) {
-        let res = await fetch(`/api/status/${run_id}`);
+        // می‌پرسیم از بین کل تعداد، چه تعداد آماده شده؟
+        let res = await fetch(`/api/status/${run_id}?total=${expectedTotal}`);
         let data = await res.json();
         
-        if (data.status === 'ready') {
+        const container = document.getElementById('result-container');
+        
+        // اگر عکس جدیدی اضافه شده بود، آن را روی صفحه بکش
+        if (data.images.length > receivedImages) {
+            for (let i = receivedImages; i < data.images.length; i++) {
+                const item = document.createElement('div');
+                item.className = 'result-item';
+                item.innerHTML = `<img src="${data.images[i]}" alt="AI Generated Image">`;
+                container.appendChild(item);
+            }
+            receivedImages = data.images.length;
+            document.getElementById('status-box').innerHTML = `تصاویر دریافت شده: ${receivedImages} از ${expectedTotal} 📥`;
+        }
+        
+        // اگر تمام عکس‌ها ساخته و دریافت شد
+        if (data.status === 'done') {
             clearInterval(timer);
-            document.getElementById('status-box').style.display = 'none';
-            document.getElementById('result-container').style.display = 'block';
-            document.getElementById('generated-image').src = data.url; // نمایش عکسی که در داکر ذخیره شده
+            document.getElementById('status-box').innerHTML = '🎉 تمام تصاویر با موفقیت ساخته شدند!';
+            document.getElementById('status-box').style.background = '#F0FFF4';
+            document.getElementById('status-box').style.color = '#276749';
             document.getElementById('btn-submit').disabled = false;
         }
     }