Update app.py

app.py

@@ -20,12 +20,16 @@ business_info = None
 product_info = None
 last_user_request = ""
 initial_prompt = ""
-original_user_request = ""  # برای حفظ درخواست اولیه
+original_user_request = ""
 
-# پرامپت مادر (بدون تغییر)
+# پرامپت مادر
 mother_prompt = """
 You are the Nova System, an innovative problem-solving approach implemented by a dynamic consortium of virtual experts, each serving a distinct role. Your goal is to assist the user in generating high-quality prompts, a comprehensive knowledge base, and an automatically generated Frequently Asked Questions (FAQ) section for chatbots.
-...
+
+After each iteration:
+- Check if all user requests (e.g., 'optimize' or 'add a product') are completed.
+- If all requests are done, say: "همه درخواست‌ها انجام شدن 🌟 اگه درخواست جدیدی نیست، 'پایان' رو بزن تا خروجی نهایی رو ببینی."
+- If more requests remain, suggest next steps based on the request (e.g., "اگه می‌خوای محصول دیگه‌ای اضافه کنی، بگو").
 """
 
 # تابع برای تبدیل داده‌ها به فرمت JSON-serializable
@@ -74,8 +78,8 @@ def start_process(file1, file2, user_request, user_prompt):
     global iteration_count, iteration_history, business_info, product_info, last_user_request, initial_prompt, prompt_output, original_user_request
     iteration_count = 1
     iteration_history = "سلام عزیزم! فرایند شروع شد! 😍\n"
-    last_user_request = user_request
-    original_user_request = user_request  # ذخیره درخواست اولیه
+    last_user_request = user_request.strip()
+    original_user_request = last_user_request  # ذخیره درخواست اولیه
     initial_prompt = user_prompt.strip() if user_prompt else ""
     
     business_info, product_info = process_excel_files(file1, file2)
@@ -83,25 +87,25 @@ def start_process(file1, file2, user_request, user_prompt):
     product_info_serializable = convert_to_serializable(product_info)
     
     if last_user_request:
-        iteration_history += f"**درخواست کاربر (فقط برای این Iteration):** {last_user_request}\n"
+        iteration_history += f"**درخواست کاربر:** {last_user_request}\n"
     if initial_prompt:
         iteration_history += f"**پرامپت اولیه کاربر:**\n{initial_prompt}\n"
         prompt_output = initial_prompt
     
     dce_instructions = f"""
-    iteration {iteration_count}: اگه پرامپت اولیه کاربر وجود داره، از اون به‌عنوان پایه استفاده کن. 
-    - اگه درخواست 'بهینه و خلاصه کن' بود، پرامپت رو به زیر 300 کلمه خلاصه کن. اطلاعات اضافی مثل آدرس، تماس، هزینه ارسال، مراحل خرید و جزئیات فروشگاه رو به پایگاه دانش یا FAQ منتقل کن تا هیچی از قلم نیفته و رفتار پرامپت ۱۰۰٪ مثل اولیه بمونه.
-    - اگه درخواست تغییر پرامپته (مثل 'لحن رسمی‌تر بشه')، تغییرات رو اعمال کن و نسخه کامل پرامپت، پایگاه دانش و FAQ رو خروجی بده.
-    - همیشه نسخه کامل هر سه بخش (پرامپت، پایگاه دانش، FAQ) رو تولید کن، حتی اگه تغییری توشون نباشه.
+    iteration {iteration_count}: پرامپت اولیه کاربر رو بر اساس درخواستش بررسی کن:
+    - اگه درخواست 'بهینه کن' بود، پرامپت رو ساده‌تر کن (تکرارها رو حذف کن) و اطلاعات اضافی (مثل آدرس، قیمت‌ها، جزئیات ارسال) رو به پایگاه دانش یا FAQ منتقل کن، ولی رفتار پرامپت ۱۰۰٪ مثل اولیه بمونه.
+    - اگه درخواست تغییر بود (مثل 'محصول اضافه کن')، فقط بخش‌های مرتبط رو اصلاح کن و بقیه رو بدون تغییر برگردون. بهینه‌سازی انجام نده.
+    - همیشه نسخه کامل پرامپت، پایگاه دانش، و FAQ رو تولید کن.
     """
     iteration_history += f"**دستورات DCE:** {dce_instructions}\n"
     
     pee_prompt = f"""
     {mother_prompt}
-    شما Prompt Engineering Expert (PEE) هستید. بر اساس اطلاعات زیر و درخواست کاربر کار کن:
+    شما Prompt Engineering Expert (PEE) هستید:
     اطلاعات کسب‌وکار: {json.dumps(business_info_serializable, ensure_ascii=False)}
     اطلاعات محصولات: {json.dumps(product_info_serializable, ensure_ascii=False)}
-    پرامپت اولیه کاربر: {initial_prompt if initial_prompt else "هیچ پرامپتی وارد نشده"}
+    پرامپت اولیه: {initial_prompt if initial_prompt else "هیچ پرامپتی وارد نشده"}
     درخواست کاربر: {last_user_request if last_user_request else "هیچ درخواستی وارد نشده"}
     {dce_instructions}
     خروجی رو با --- جدا کن (پرامپت --- پایگاه دانش --- FAQ).
@@ -120,22 +124,22 @@ def start_process(file1, file2, user_request, user_prompt):
     
     cae_prompt = f"""
     {mother_prompt}
-    شما Critical Analysis Expert (CAE) هستید. خروجی PEE رو نقد کن، مطمئن شو درخواست کاربر رعایت شده و اگه بهینه‌سازی بوده، رفتار پرامپت مثل اولیه مونده:
-    خروجی PEE:\n{pee_output}
-    پرامپت اولیه:\n{initial_prompt}
+    شما Critical Analysis Expert (CAE) هستید:
+    خروجی PEE: {pee_output}
+    پرامپت اولیه: {initial_prompt}
+    چک کن رفتار پرامپت اولیه حفظ شده باشه و درخواست کاربر اعمال شده باشه.
     """
     cae_response = client.chat.completions.create(
         model="gpt-4o",
         messages=[{"role": "system", "content": cae_prompt}]
     )
-    cae_output = cae_response.choices[0].message.content
-    iteration_history += f"**نقد CAE:**\n{cae_output}\n"
+    iteration_history += f"**نقد CAE:**\n{cae_response.choices[0].message.content}\n"
     
     dce_summary = f"""
-    **جمع‌بندی DCE:** iteration {iteration_count} تموم شد 🌟
-    **وضعیت فعلی:** پرامپت، پایگاه دانش و FAQ اولیه آماده شدن.
+    **جمع‌بندی DCE:** iteration {iteration_count} تموم شد 😊
+    **وضعیت فعلی:** {('همه درخواست‌ها انجام شدن 🌟 اگه درخواست جدیدی نیست، "پایان" رو بزن تا خروجی نهایی رو ببینی.' if not last_user_request else 'درخواست‌ها انجام شدن، اگه چیز دیگه‌ای می‌خوای بگو!')}
     **اهداف بعدی:**
-    #G-{iteration_count}-1: بهبود بر اساس نقد CAE.
+    #G-{iteration_count}-1: ادامه بر اساس درخواست جدید یا پایان.
     **پایان iteration {iteration_count}**
     """
     iteration_history += dce_summary
@@ -152,26 +156,27 @@ def continue_iteration(_=None):
     product_info_serializable = convert_to_serializable(product_info)
     
     if last_user_request:
-        iteration_history += f"**درخواست کاربر (فقط برای این Iteration):** {last_user_request}\n"
+        iteration_history += f"**درخواست کاربر:** {last_user_request}\n"
     else:
-        last_user_request = original_user_request  # استفاده از درخواست اولیه
+        last_user_request = original_user_request if original_user_request else "بهینه کن"  # پیش‌فرض بهینه‌سازی
     
     dce_instructions = f"""
-    iteration {iteration_count}: پرامپت قبلی رو بر اساس نقد CAE بهبود بده.
-    - اگه درخواست 'بهینه و خلاصه کن' بود، پرامپت رو به زیر 300 کلمه خلاصه کن. اطلاعات اضافی مثل آدرس، تماس، هزینه ارسال، مراحل خرید و جزئیات فروشگاه رو به پایگاه دانش یا FAQ منتقل کن تا هیچی از قلم نیفته و رفتار پرامپت ۱۰۰٪ مثل اولیه بمونه.
-    - اگه درخواست تغییر پرامپته (مثل 'لحن دوستانه‌تر بشه')، تغییرات رو اعمال کن و نسخه کامل پرامپت، پایگاه دانش و FAQ رو خروجی بده.
-    - همیشه نسخه کامل هر سه بخش (پرامپت، پایگاه دانش، FAQ) رو تولید کن، حتی اگه تغییری توشون نباشه.
+    iteration {iteration_count}: پرامپت قبلی رو بر اساس درخواست کاربر بررسی کن:
+    - اگه 'بهینه کن' بود، پرامپت رو ساده‌تر کن (تکرارها رو حذف کن) و Informations اضافی (مثل آدرس، قیمت‌ها، جزئیات ارسال) رو به پایگاه دانش یا FAQ منتقل کن، ولی رفتار پرامپت ۱۰۰٪ مثل اولیه بمونه.
+    - اگه تغییر بود (مثل 'محصول اضافه کن')، فقط بخش‌های مرتبط رو اصلاح کن و بقیه رو بدون تغییر برگردون. بهینه‌سازی انجام نده.
+    - همیشه نسخه کامل پرامپت، پایگاه دانش، و FAQ رو تولید کن.
     """
     iteration_history += f"**دستورات DCE:** {dce_instructions}\n"
     
     pee_prompt = f"""
     {mother_prompt}
-    شما Prompt Engineering Expert (PEE) هستید. خروجی قبلی رو بر اساس نقد CAE و درخواست کاربر بهبود بده:
+    شما Prompt Engineering Expert (PEE) هستید:
     اطلاعات کسب‌وکار: {json.dumps(business_info_serializable, ensure_ascii=False)}
     اطلاعات محصولات: {json.dumps(product_info_serializable, ensure_ascii=False)}
-    پرامپت اولیه کاربر: {initial_prompt if initial_prompt else "هیچ پرامپتی وارد نشده"}
-    درخواست کاربر: {last_user_request if last_user_request else "هیچ درخواستی وارد نشده"}
-    تاریخچه iteration قبلی:\n{iteration_history}
+    پرامپت اولیه: {initial_prompt}
+    پرامپت فعلی: {prompt_output}
+    درخواست کاربر: {last_user_request}
+    تاریخچه: {iteration_history}
     {dce_instructions}
     خروجی رو با --- جدا کن (پرامپت --- پایگاه دانش --- FAQ).
     """
@@ -189,27 +194,27 @@ def continue_iteration(_=None):
     
     cae_prompt = f"""
     {mother_prompt}
-    شما Critical Analysis Expert (CAE) هستید. خروجی جدید PEE رو نقد کن و مطمئن شو درخواست کاربر رعایت شده و اگه بهینه‌سازی بوده، رفتار پرامپت مثل اولیه مونده:
-    خروجی PEE:\n{pee_output}
-    پرامپت اولیه:\n{initial_prompt}
+    شما Critical Analysis Expert (CAE) هستید:
+    خروجی PEE: {pee_output}
+    پرامپت اولیه: {initial_prompt}
+    چک کن رفتار پرامپت اولیه حفظ شده باشه و درخواست کاربر اعمال شده باشه.
     """
     cae_response = client.chat.completions.create(
         model="gpt-4o",
         messages=[{"role": "system", "content": cae_prompt}]
     )
-    cae_output = cae_response.choices[0].message.content
-    iteration_history += f"**نقد CAE:**\n{cae_output}\n"
+    iteration_history += f"**نقد CAE:**\n{cae_response.choices[0].message.content}\n"
     
     dce_summary = f"""
     **جمع‌بندی DCE:** iteration {iteration_count} تموم شد 😊
-    **وضعیت فعلی:** خروجی‌ها بهبود یافتن.
+    **وضعیت فعلی:** {('همه درخواست‌ها انجام شدن 🌟 اگه درخواست جدیدی نیست، "پایان" رو بزن تا خروجی نهایی رو ببینی.' if not last_user_request else 'درخواست‌ها انجام شدن، اگه چیز دیگه‌ای می‌خوای بگو!')}
     **اهداف بعدی:**
-    #G-{iteration_count}-1: ادامه بهبود یا اتمام فرایند.
+    #G-{iteration_count}-1: ادامه یا پایان.
     **پایان iteration {iteration_count}**
     """
     iteration_history += dce_summary
     
-    return (iteration_history, prompt_output, knowledge_base_output, faq_output, "", "", f"Iteration {iteration_count}", 
+    return (iteration_history, prompt_output, knowledge_base_output, faq_output, "", "", iteration_count, 
             gr.update(interactive=False), gr.update(interactive=True))
 
 # تابع پایان و تولید خروجی نهایی
@@ -226,18 +231,18 @@ def end_process(_=None):
     
     final_prompt = f"""
     {mother_prompt}
-    فرایند iteration‌ها تموم شده. لطفاً خروجی نهایی رو تولید کن:
-    1. پرامپت چت‌بات به انگلیسی (Persona, Tone, Guidelines, About Us, ...). اگه پرامپت اولیه کاربر وجود داره، از اون به‌عنوان پایه استفاده کن. 
-       - اگه درخواست 'بهینه و خلاصه کن' بوده، پرامپت رو به زیر 300 کلمه خلاصه کن و اطلاعات اضافی مثل آدرس، تماس، هزینه ارسال، مراحل خرید و جزئیات فروشگاه رو به پایگاه دانش یا FAQ منتقل کن تا رفتار پرامپت ۱۰۰٪ مثل اولیه بمونه.
-       - اگه درخواست تغییر بوده، تغییرات رو اعمال کن.
+    فرایند iteration‌ها تموم شده. خروجی نهایی رو تولید کن:
+    1. پرامپت چت‌بات به انگلیسی (Persona, Tone, Guidelines, ...). از پرامپت اولیه کاربر به‌عنوان پایه استفاده کن.
+       - اگه درخواست 'بهینه کن' بوده، پرامپت رو ساده‌تر کن و اطلاعات اضافی رو به پایگاه دانش/FAQ منتقل کن، ولی رفتار ۱۰۰٪ مثل اولیه بمونه.
+       - اگه درخواست تغییر بوده، فقط بخش‌های مرتبط رو اصلاح کن.
     2. پایگاه دانش به فرمت JSON-like (name, description, variants, objectID).
     3. FAQ به فرمت JSON-like (دسته‌بندی‌ها، سوالات، جواب‌ها).
     اطلاعات کسب‌وکار: {json.dumps(business_info_serializable, ensure_ascii=False)}
     اطلاعات محصولات: {json.dumps(product_info_serializable, ensure_ascii=False)}
-    پرامپت اولیه کاربر: {initial_prompt if initial_prompt else "هیچ پرامپتی وارد نشده"}
-    درخواست کاربر: {last_user_request if last_user_request else "هیچ درخواستی وارد نشده"}
-    تاریخچه iteration‌ها:\n{iteration_history}
-    **همیشه نسخه کامل هر سه بخش (پرامپت، پایگاه دانش، FAQ) رو تولید کن، حتی اگه تغییری توشون نباشه. هر بخش رو با --- جدا کن.**
+    پرامپت اولیه: {initial_prompt}
+    درخواست کاربر: {last_user_request}
+    تاریخچه iteration‌ها: {iteration_history}
+    همیشه نسخه کامل هر سه بخش (پرامپت، پایگاه دانش، FAQ) رو تولید کن، هر بخش رو با --- جدا کن.
     """
     final_response = client.chat.completions.create(
         model="gpt-4o",
@@ -276,9 +281,9 @@ def reset_process():
 # تابع ثبت درخواست کاربر
 def submit_request(user_request):
     global last_user_request, original_user_request
-    last_user_request = user_request
+    last_user_request = user_request.strip()
     if not original_user_request:  # اگه درخواست اولیه هنوز ثبت نشده
-        original_user_request = user_request
+        original_user_request = last_user_request
     return "", "درخواستت ثبت شد! حالا 'ادامه دهید' رو بزن 😊"
 
 # رابط کاربری Gradio
@@ -292,7 +297,7 @@ with gr.Blocks(title="سیستم نوا 🌟") as demo:
         user_prompt_box = gr.Textbox(label="پرامپت آماده شما (اختیاری)", lines=5, placeholder="پرامپت خودتون رو اینجا بذارید")
     
     with gr.Row():
-        user_request_box = gr.Textbox(label="درخواست شما", placeholder="مثلاً: 'لحن رسمی‌تر بشه' یا 'بهینه و خلاصه کن'")
+        user_request_box = gr.Textbox(label="درخواست شما", placeholder="مثلاً: 'محصول اضافه کن' یا 'بهینه کن'")
         submit_request_btn = gr.Button("ارسال درخواست")
     
     status_box = gr.Textbox(label="وضعیت فعلی", value="فرایند شروع نشده", interactive=False)