Update app.py

app.py

@@ -18,7 +18,7 @@ knowledge_base_output = ""
 faq_output = ""
 business_info = None
 product_info = None
-last_user_request = ""  # متغیر جدید برای ذخیره آخرین درخواست کاربر
+last_user_request = ""
 
 # پرامپت مادر (بدون تغییر)
 mother_prompt = """
@@ -42,11 +42,20 @@ def validate_and_fix_output(output):
     parts = output.split("---")
     parts = [part.strip() for part in parts if part.strip()]
     print("بخش‌های جدا شده:", parts)
-    if len(parts) > 3 and not parts[0].startswith("**Persona:") and not parts[0].startswith("["):
-        parts = parts[1:]
+    
+    # مطمئن می‌شیم که همیشه سه بخش برگردونده بشه
     prompt_part = parts[0] if len(parts) > 0 else "پرامپت تولید نشد 😔"
     kb_part = parts[1] if len(parts) > 1 else "پایگاه دانش تولید نشد 😕"
     faq_part = parts[2] if len(parts) > 2 else "FAQ تولید نشد 🥳"
+    
+    # اگه بخش‌ها ناقص باشن، یه پیام پیش‌فرض بذاریم
+    if not prompt_part.strip():
+        prompt_part = "پرامپت خالیه 😔"
+    if not kb_part.strip():
+        kb_part = "پایگاه دانش خالیه 😕"
+    if not faq_part.strip():
+        faq_part = "FAQ خالیه 🥳"
+    
     return [prompt_part, kb_part, faq_part]
 
 # تابع برای خوندن و پردازش فایل‌های اکسل
@@ -61,7 +70,7 @@ def start_process(file1, file2, user_request):
     global iteration_count, iteration_history, business_info, product_info, last_user_request
     iteration_count = 1
     iteration_history = "سلام عزیزم! فرایند شروع شد! 😍\n"
-    last_user_request = user_request  # ذخیره درخواست اولیه
+    last_user_request = user_request
     
     business_info, product_info = process_excel_files(file1, file2)
     business_info_serializable = convert_to_serializable(business_info)
@@ -110,13 +119,13 @@ def start_process(file1, file2, user_request):
     **پایان iteration {iteration_count}**
     """
     iteration_history += dce_summary
-    last_user_request = ""  # خالی کردن درخواست بعد از اعمال
+    last_user_request = ""
     
     return (iteration_history, "", "", "", "", f"Iteration {iteration_count}", 
             gr.update(interactive=False), gr.update(interactive=True))
 
 # تابع ادامه Iteration
-def continue_iteration(_=None):  # ورودی رو اختیاری کردیم چون از last_user_request استفاده می‌کنیم
+def continue_iteration(_=None):
     global iteration_count, iteration_history, prompt_output, knowledge_base_output, faq_output, last_user_request
     iteration_count += 1
     
@@ -171,13 +180,13 @@ def continue_iteration(_=None):  # ورودی رو اختیاری کردیم چ
     **پایان iteration {iteration_count}**
     """
     iteration_history += dce_summary
-    last_user_request = ""  # خالی کردن درخواست بعد از اعمال
+    last_user_request = ""
     
     return (iteration_history, prompt_output, knowledge_base_output, faq_output, "", 
             f"Iteration {iteration_count}", gr.update(interactive=False), gr.update(interactive=True))
 
 # تابع پایان و تولید خروجی نهایی
-def end_process(_=None):  # ورودی رو اختیاری کردیم
+def end_process(_=None):
     global iteration_history, prompt_output, knowledge_base_output, faq_output, last_user_request
     
     business_info_serializable = convert_to_serializable(business_info)
@@ -220,8 +229,13 @@ def end_process(_=None):  # ورودی رو اختیاری کردیم
     knowledge_base_output = parts[1]
     faq_output = parts[2]
     
+    # لاگ اضافی برای چک کردن متغیرها
+    print("پرامپت نهایی:", prompt_output)
+    print("پایگاه دانش نهایی:", knowledge_base_output)
+    print("FAQ نهایی:", faq_output)
+    
     iteration_history += "\n**فرایند تموم شد و خروجی نهایی آماده‌ست! 🎉**\n"
-    last_user_request = ""  # خالی کردن درخواست بعد از اعمال
+    last_user_request = ""
     
     return (iteration_history, prompt_output, knowledge_base_output, faq_output, "", 
             "فرایند پایان یافت", gr.update(interactive=True), gr.update(interactive=True))
@@ -288,17 +302,4 @@ with gr.Blocks(title="سیستم نوا 🌟") as demo:
     end_btn.click(
         end_process,
         inputs=[],
-        outputs=[iteration_box, prompt_box, kb_box, faq_box, user_request_box, status_box, start_btn, continue_btn]
-    )
-    reset_btn.click(
-        reset_process,
-        inputs=[],
-        outputs=[iteration_box, prompt_box, kb_box, faq_box, user_request_box, status_box, start_btn, continue_btn]
-    )
-    submit_request_btn.click(
-        submit_request,
-        inputs=[user_request_box],
-        outputs=[user_request_box, status_box]
-    )
-
-demo.launch()
+        outputs=[iteration_box, prompt_box, kb_box, faq_box, user_request_box, status_box, start