Update app.py

app.py

@@ -6,14 +6,14 @@ from werkzeug.security import generate_password_hash, check_password_hash
 import json
 import time
 
-# تحميل نموذج Stable Diffusion الأخف
+# تحميل نموذج Stable Diffusion
 MODEL_NAME = "runwayml/stable-diffusion-v1-5"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = StableDiffusionPipeline.from_pretrained(MODEL_NAME)
 pipe.to(device)
 
-# قاعدة بيانات للمستخدمين والنماذج المحفوظة (سيتم حفظها في ملف JSON)
+# قاعدة بيانات للمستخدمين والنماذج المحفوظة
 user_data_file = "user_data.json"
 
 # تحميل البيانات من ملف JSON إذا كان موجودًا
@@ -25,18 +25,16 @@ else:
 
 saved_models = {}
 
-# وظيفة تدريب النموذج (التدريب يتم مرة واحدة فقط)
-def train_model(user_email, images, progress):
+# وظيفة تدريب النموذج (تمت إزالة شريط التقدم)
+def train_model(user_email, images):
     if not user_email:
-        return "يجب تسجيل الدخول لحفظ النموذج."
+        return "❌ يجب تسجيل الدخول لحفظ النموذج."
     
     user_model_id = f"user_model_{user_email}"
     saved_models[user_email] = user_model_id
 
-    # محاكاة عملية التدريب (يمكنك تعديل هذا الجزء ليتناسب مع عملية التدريب الفعلية)
-    for i in range(1, 101):  # محاكاة من 0% إلى 100%
-        time.sleep(0.1)  # تأخير زمني لمحاكاة الوقت
-        progress(i)  # تحديث شريط التقدم
+    # محاكاة عملية التدريب
+    time.sleep(5)  # تأخير زمني لمحاكاة التدريب (يمكن تعديله)
 
     return f"✅ تم حفظ النموذج بنجاح: {user_model_id}"
 
@@ -79,14 +77,13 @@ with gr.Blocks() as demo:
             user_email_input = gr.Textbox(label="📧 بريدك الإلكتروني (اختياري لحفظ النموذج)")
             train_button = gr.Button("🔧 تدريب النموذج")
             train_output = gr.Textbox(label="🔔 نتيجة التدريب")
-            progress_bar = gr.Progress()  # بدون label هنا
  
         with gr.Column():
             prompt_input = gr.Textbox(label="✏️ أدخل البرومبت لإنشاء صورة")
             generate_button = gr.Button("🎨 إنشاء صورة")
             output_image = gr.Image(label="📷 الصورة الناتجة")
  
-    train_button.click(train_model, inputs=[user_email_input, image_input, progress_bar], outputs=train_output)
+    train_button.click(train_model, inputs=[user_email_input, image_input], outputs=train_output)
     generate_button.click(generate_image, inputs=[prompt_input, user_email_input], outputs=output_image)
  
     gr.Markdown("### 🔑 تسجيل الدخول / التسجيل")
@@ -108,4 +105,4 @@ with gr.Blocks() as demo:
     register_button.click(register, inputs=[register_email, register_password], outputs=register_output)
 
 # تشغيل التطبيق مع رابط عام
-demo.launch(share=True)
+demo.launch(share=True)