Update app.py

app.py

@@ -1,29 +1,73 @@
+import os
 import gradio as gr
+import torch
+from diffusers import StableDiffusionPipeline
+from werkzeug.security import generate_password_hash, check_password_hash
+import json
 import time
 
-# محاكاة وظيفة تسجيل الدخول
-def login(email, password):
-    if email == "[email protected]" and password == "password":
-        return "تم تسجيل الدخول بنجاح!"
-    else:
-        return "البريد الإلكتروني أو كلمة المرور غير صحيحة."
+# تحميل نموذج Stable Diffusion الأخف
+MODEL_NAME = "runwayml/stable-diffusion-v1-5"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# محاكاة وظيفة التسجيل
-def register(email, password):
-    return f"تم تسجيل حساب جديد للبريد الإلكتروني: {email}"
+pipe = StableDiffusionPipeline.from_pretrained(MODEL_NAME)
+pipe.to(device)
+
+# قاعدة بيانات للمستخدمين والنماذج المحفوظة (سيتم حفظها في ملف JSON)
+user_data_file = "user_data.json"
+
+# تحميل البيانات من ملف JSON إذا كان موجودًا
+if os.path.exists(user_data_file):
+    with open(user_data_file, "r") as f:
+        users = json.load(f)
+else:
+    users = {}
+
+saved_models = {}
+
+# وظيفة تدريب النموذج (التدريب يتم مرة واحدة فقط)
+def train_model(user_email, images, progress):
+    if not user_email:
+        return "يجب تسجيل الدخول لحفظ النموذج."
+    
+    user_model_id = f"user_model_{user_email}"
+    saved_models[user_email] = user_model_id
 
-# محاكاة وظيفة التدريب
-def train_model(images, progress):
     # محاكاة عملية التدريب (يمكنك تعديل هذا الجزء ليتناسب مع عملية التدريب الفعلية)
     for i in range(1, 101):  # محاكاة من 0% إلى 100%
         time.sleep(0.1)  # تأخير زمني لمحاكاة الوقت
-        progress(f"{i}%")  # تحديث التقدم النصي
+        progress(i)  # تحديث شريط التقدم
 
-    return "✅ تم حفظ النموذج بنجاح"
+    return f"✅ تم حفظ النموذج بنجاح: {user_model_id}"
 
-# محاكاة وظيفة إنشاء الصورة
-def generate_image(prompt):
-    return f"📸 تم إنشاء صورة بناءً على البرومبت: {prompt}"
+# وظيفة إنشاء الصور باستخدام البرومبتات
+def generate_image(prompt, user_email=None):
+    if user_email and user_email in saved_models:
+        model_id = saved_models[user_email]
+        result = pipe(prompt).images[0]
+    else:
+        result = pipe(prompt).images[0]
+
+    return result
+
+# وظيفة تسجيل المستخدمين
+def register(email, password):
+    if email in users:
+        return "⚠️ البريد الإلكتروني مسجل بالفعل!"
+    users[email] = {"password": generate_password_hash(password)}
+    save_users_data()
+    return "✅ تم التسجيل بنجاح!"
+
+# وظيفة تسجيل الدخول
+def login(email, password):
+    if email not in users or not check_password_hash(users[email]["password"], password):
+        return "❌ البريد الإلكتروني أو كلمة المرور غير صحيحة!"
+    return f"✅ تم تسجيل الدخول: {email}"
+
+# حفظ بيانات المستخدمين إلى ملف JSON
+def save_users_data():
+    with open(user_data_file, "w") as f:
+        json.dump(users, f)
 
 # واجهة Gradio
 with gr.Blocks() as demo:
@@ -31,18 +75,19 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column():
-            image_input = gr.Files(label="📤 رفع صورك للتدريب")
+            image_input = gr.Files(label="📤 رفع صورك للتدريب")  # ✅ تعديل هنا
+            user_email_input = gr.Textbox(label="📧 بريدك الإلكتروني (اختياري لحفظ النموذج)")
             train_button = gr.Button("🔧 تدريب النموذج")
             train_output = gr.Textbox(label="🔔 نتيجة التدريب")
-            progress_bar = gr.Textbox(label="🔄 التقدم", interactive=False)  # تحديث التقدم هنا
+            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=[image_input, progress_bar], outputs=train_output)
-    generate_button.click(generate_image, inputs=[prompt_input], outputs=output_image)
+    train_button.click(train_model, inputs=[user_email_input, image_input, progress_bar], outputs=train_output)
+    generate_button.click(generate_image, inputs=[prompt_input, user_email_input], outputs=output_image)
  
     gr.Markdown("### 🔑 تسجيل الد��ول / التسجيل")
     
@@ -61,6 +106,6 @@ with gr.Blocks() as demo:
  
     login_button.click(login, inputs=[login_email, login_password], outputs=login_output)
     register_button.click(register, inputs=[register_email, register_password], outputs=register_output)
- 
-# تشغيل التطبيق
-demo.launch()
+
+# تشغيل التطبيق مع رابط عام
+demo.launch(share=True)