Ahmed1871992 commited on
Commit
4288ff5
·
verified ·
1 Parent(s): 2cd60b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -14
app.py CHANGED
@@ -13,24 +13,16 @@ def register(email, password):
13
  return f"تم تسجيل حساب جديد للبريد الإلكتروني: {email}"
14
 
15
  # محاكاة وظيفة التدريب
16
- def train_model(user_email, images, progress):
17
- if not user_email:
18
- return "يجب تسجيل الدخول لحفظ النموذج."
19
-
20
- user_model_id = f"user_model_{user_email}"
21
- saved_models[user_email] = user_model_id
22
-
23
  # محاكاة عملية التدريب (يمكنك تعديل هذا الجزء ليتناسب مع عملية التدريب الفعلية)
24
  for i in range(1, 101): # محاكاة من 0% إلى 100%
25
  time.sleep(0.1) # تأخير زمني لمحاكاة الوقت
26
  progress(f"{i}%") # تحديث التقدم النصي
27
 
28
- return f"✅ تم حفظ النموذج بنجاح: {user_model_id}"
29
 
30
  # محاكاة وظيفة إنشاء الصورة
31
- def generate_image(prompt, user_email):
32
- if not user_email:
33
- return "يجب تسجيل الدخول لإنشاء صورة."
34
  return f"📸 تم إنشاء صورة بناءً على البرومبت: {prompt}"
35
 
36
  # واجهة Gradio
@@ -40,7 +32,6 @@ with gr.Blocks() as demo:
40
  with gr.Row():
41
  with gr.Column():
42
  image_input = gr.Files(label="📤 رفع صورك للتدريب")
43
- user_email_input = gr.Textbox(label="📧 بريدك الإلكتروني (اختياري لحفظ النموذج)")
44
  train_button = gr.Button("🔧 تدريب النموذج")
45
  train_output = gr.Textbox(label="🔔 نتيجة التدريب")
46
  progress_bar = gr.Textbox(label="🔄 التقدم", interactive=False) # تحديث التقدم هنا
@@ -50,8 +41,8 @@ with gr.Blocks() as demo:
50
  generate_button = gr.Button("🎨 إنشاء صورة")
51
  output_image = gr.Image(label="📷 الصورة الناتجة")
52
 
53
- train_button.click(train_model, inputs=[user_email_input, image_input, progress_bar], outputs=train_output)
54
- generate_button.click(generate_image, inputs=[prompt_input, user_email_input], outputs=output_image)
55
 
56
  gr.Markdown("### 🔑 تسجيل الدخول / التسجيل")
57
 
 
13
  return f"تم تسجيل حساب جديد للبريد الإلكتروني: {email}"
14
 
15
  # محاكاة وظيفة التدريب
16
+ def train_model(images, progress):
 
 
 
 
 
 
17
  # محاكاة عملية التدريب (يمكنك تعديل هذا الجزء ليتناسب مع عملية التدريب الفعلية)
18
  for i in range(1, 101): # محاكاة من 0% إلى 100%
19
  time.sleep(0.1) # تأخير زمني لمحاكاة الوقت
20
  progress(f"{i}%") # تحديث التقدم النصي
21
 
22
+ return "✅ تم حفظ النموذج بنجاح"
23
 
24
  # محاكاة وظيفة إنشاء الصورة
25
+ def generate_image(prompt):
 
 
26
  return f"📸 تم إنشاء صورة بناءً على البرومبت: {prompt}"
27
 
28
  # واجهة Gradio
 
32
  with gr.Row():
33
  with gr.Column():
34
  image_input = gr.Files(label="📤 رفع صورك للتدريب")
 
35
  train_button = gr.Button("🔧 تدريب النموذج")
36
  train_output = gr.Textbox(label="🔔 نتيجة التدريب")
37
  progress_bar = gr.Textbox(label="🔄 التقدم", interactive=False) # تحديث التقدم هنا
 
41
  generate_button = gr.Button("🎨 إنشاء صورة")
42
  output_image = gr.Image(label="📷 الصورة الناتجة")
43
 
44
+ train_button.click(train_model, inputs=[image_input, progress_bar], outputs=train_output)
45
+ generate_button.click(generate_image, inputs=[prompt_input], outputs=output_image)
46
 
47
  gr.Markdown("### 🔑 تسجيل الدخول / التسجيل")
48