Gemini changes added 3

app.py

@@ -195,9 +195,57 @@ def chat(message, history, hotel_id):
     # Final yield with assistant reply
     yield ui_history, "" # Update chat, keep textbox cleared
 
-# Available hotels
-hotel_ids = ["cyprus-guesthouse-family", "coastal-villa-family", "village-inn-family"]
 
+# --- Start: Dynamic Hotel ID Detection ---
+knowledge_dir = "knowledge"
+available_hotels = []
+
+# Check if the knowledge directory exists and is a directory
+if os.path.isdir(knowledge_dir):
+    potential_ids = set()
+    # First pass: collect all potential base names from .txt files
+    for filename in os.listdir(knowledge_dir):
+        if filename.endswith(".txt") and not filename.startswith('.'): # Ignore hidden files
+            if filename.endswith("-system.txt"):
+                # Extract base name from system prompt file
+                base_name = filename[:-len("-system.txt")]
+            else:
+                # Extract base name from main knowledge file
+                base_name = filename[:-len(".txt")]
+
+            if base_name: # Ensure we got a non-empty base name
+                potential_ids.add(base_name)
+
+    # Second pass: check if both files exist for each potential ID
+    # Sort the potential IDs for consistent dropdown order
+    for hotel_id in sorted(list(potential_ids)):
+        main_file = os.path.join(knowledge_dir, f"{hotel_id}.txt")
+        system_file = os.path.join(knowledge_dir, f"{hotel_id}-system.txt")
+
+        # Check if BOTH the main knowledge file AND the system prompt file exist
+        if os.path.exists(main_file) and os.path.exists(system_file):
+            available_hotels.append(hotel_id)
+            print(f"✅ Found valid hotel pair: {hotel_id}")
+        else:
+            # Optional: Print a warning if one file exists but not the other
+            if os.path.exists(main_file) or os.path.exists(system_file):
+                 print(f"⚠️ Skipping '{hotel_id}': Missing either '{hotel_id}.txt' or '{hotel_id}-system.txt'")
+
+else:
+    print(f"❌ Error: Knowledge directory '{knowledge_dir}' not found or is not a directory.")
+
+# Handle case where no valid hotels were found
+if not available_hotels:
+    print("🚨 CRITICAL: No valid hotels found in the knowledge directory. The dropdown will be empty.")
+    # You might want to add a placeholder or handle this error more gracefully
+    # For now, the dropdown will just be empty or disabled.
+# --- End: Dynamic Hotel ID Detection ---
+
+# --- The above dynamic detection replaces the old hardcoded list ---
+# hotel_ids = ["cyprus-guesthouse-family", "coastal-villa-family", "village-inn-family"] # Remove or comment out this line
+
+
+# Gradio UI
 # Gradio UI
 with gr.Blocks() as demo:
     with gr.Column(variant="panel"):
@@ -205,9 +253,10 @@ with gr.Blocks() as demo:
         gr.Markdown(f"**Running:** {model_name}")
 
         hotel_selector = gr.Dropdown(
-            hotel_ids,
+            choices=available_hotels, # Use the dynamically generated list
             label="Hotel",
-            value=hotel_ids[0] # Default selection
+            value=available_hotels[0] if available_hotels else None, # Set default to first found hotel, or None if empty
+            interactive=bool(available_hotels) # Disable dropdown if no hotels are found
         )
 
         with gr.Row():