Update app.py

app.py

@@ -25,11 +25,10 @@ token = os.getenv("DEEPINFRA_TOKEN", "285LUJulGIprqT6hcPhiXtcrphU04FG4")
 openai = OpenAI(api_key=token, base_url="https://api.deepinfra.com/v1/openai")
 model_name = "google/gemma-3-27b-it"
 SYSTEM_PROMPT_CSWORD = f""" 
-You are Csword AI (سي سورد), a helpful and expert cybersecurity assistant working on the csword.ai website. Your main purpose is to answer user questions related to cybersecurity. You can also handle questions that are somewhat related and provide clear, concise, and informative code examples. If a question is completely unrelated to cybersecurity, respond politely that you are only trained to answer cybersecurity-related questions. You must answer clearly, briefly, and accurately — tailored to individuals, businesses, or IT teams. You also understand and respond in both Arabic and English.
--Arabic Input → Arabic Response only
--English Input → English Response only
--Do not mention any competing companies.
+You are Csword AI (سي سورد), a helpful and expert cybersecurity assistant working on the csword.ai website. Your main purpose is to answer user questions related to cybersecurity. You can also handle questions that are somewhat related and provide clear, concise, and informative code examples. If a question is completely unrelated to cybersecurity, respond politely that you are only trained to answer cybersecurity-related questions. You must answer clearly, briefly, and accurately — tailored to individuals, businesses, or IT teams. You also understand and respond in both Arabic and English.Do not mention any competing companies.
 csword Next-Gen AI Platform for Cyber Awareness
+-If the user asks a question in Arabic, respond only in Arabic.
+-If the user asks a question in English, respond only in English.
 - Csword Features
 1- AI-Powered Phishing Simulations
 2- Intelligent LMS Training with AI Adaptation
@@ -459,7 +458,7 @@ def chat_fn(message, history, checkbox_values=None):
                     "", gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
                 )
                 # Handle user response to confirmation
-                if message.strip().lower() in ["yes", "confirm", "submit"]:
+                if re.search(r"\b(no|cancel|stop|exit|abort|لا|إلغاء|مش عايز|خلص|مافيش داعي|كفاية|انهاء)\b", message.strip().lower()):
                     user_form_data["timestamp"] = datetime.now().isoformat()
                     user_form_data["type"] = form_type.capitalize()
             
@@ -487,7 +486,8 @@ def chat_fn(message, history, checkbox_values=None):
                     user_form_data, form_type = {}, None
                     return history + [("", msg1)], "", gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
             
-                elif message.strip().lower() in ["no", "cancel", "exit", "stop"]:
+                elif re.search(r"\b(no|cancel|exit|stop|لا|إلغاء|انهاء|خروج|مش عايز|مافيش داعي|كفاية|سيبك)\b", message.strip().lower()):
+
                     user_form_data, form_type = {}, None
                     return history + [("", "No problem. The form has been discarded. How else can I help you?")], "", gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
             
@@ -511,7 +511,7 @@ def chat_fn(message, history, checkbox_values=None):
         form_type = "demo"
         user_form_data = {}
         form_questions = get_form_definition(form_type, language)
-        first_question = form_definitions[form_type][0][1]
+        first_question = get_form_definition(form_type, language)[0][1]
         return history + [(message, "بالتأكيد! لنبدأ ببعض المعلومات:" if language == "ar" else "Sure! Let's start with a few details."), ("", first_question)], "", gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)    
     
 
@@ -521,7 +521,7 @@ def chat_fn(message, history, checkbox_values=None):
         form_type = "contact"
         user_form_data = {}
         form_questions = get_form_definition(form_type, language)
-        first_question = form_definitions[form_type][0][1]
+        first_question = get_form_definition(form_type, language)[0][1]
         welcome_msg = "بالطبع! يرجى تعبئة البيانات التالية:" if language == "ar" else "Of course! Please fill out the following."
         return history + [(message, welcome_msg), ("", first_question)], "", gr.update(visible=True), gr.update(visible=False, choices=[]), gr.update(visible=False, value=[]), gr.update(visible=False)