Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import logging
+from difflib import get_close_matches  # For fuzzy matching
 
 # Omdurman National Bank-specific guidelines
 ONB_GUIDELINES = {
@@ -34,13 +35,32 @@ def respond(message: str, history: list):
     # Log the query
     logging.info(f"Query: {message}")
     
-    # Check for keywords in the message
-    for keyword, key in KEYWORD_TO_RESPONSE.items():
+    # Check for "Back to Menu" keyword
+    if "القائمة" in message or "menu" in message.lower():
+        return "main_menu"
+    
+    # Find all matching keywords using fuzzy matching
+    matches = []
+    for keyword in KEYWORD_TO_RESPONSE.keys():
         if keyword in message:
-            return ONB_GUIDELINES.get(key, "عذرًا، لم يتم التعرف على الخيار المحدد.")
+            matches.append(keyword)
+        else:
+            # Use fuzzy matching for similar words
+            close_matches = get_close_matches(keyword, message.split(), n=1, cutoff=0.6)
+            if close_matches:
+                matches.append(keyword)
+    
+    # If multiple matches are found, return all responses
+    if matches:
+        responses = []
+        for match in matches:
+            key = KEYWORD_TO_RESPONSE.get(match)
+            if key:
+                responses.append(ONB_GUIDELINES.get(key))
+        return "\n\n".join(responses)
     
     # Default response if no keyword is matched
-    return "عذرًا، لم أفهم سؤالك. الرجاء إعادة الصياغة أو اختيار أحد الخيارات التالية: " + ", ".join(KEYWORD_TO_RESPONSE.keys())
+    return "عذرًا، لم أفهم سؤالك. الرجاء إعادة الصياغة أو كتابة 'القائمة' للعودة إلى القائمة الرئيسية."
 
 # Main menu with submenus
 main_menu = {