Update app.py

app.py

@@ -5,16 +5,21 @@ import markdown
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# Function to read and process Markdown files from the 'data' directory
+# Function to read and process Markdown files from the 'data' directory and its subfolders
 def load_markdown_files(data_folder='data'):
     documents = []
-    for filename in os.listdir(data_folder):
-        if filename.endswith('.md'):
-            with open(os.path.join(data_folder, filename), 'r', encoding='utf-8') as file:
-                content = file.read()
-                # Convert Markdown to plain text if needed
-                html_content = markdown.markdown(content)
-                documents.append(html_content)  # Store HTML content or plain text
+    for root, dirs, files in os.walk(data_folder):
+        for filename in files:
+            if filename.endswith('.md'):
+                file_path = os.path.join(root, filename)
+                try:
+                    with open(file_path, 'r', encoding='utf-8') as file:
+                        content = file.read()
+                        # Convert Markdown to plain text if needed
+                        html_content = markdown.markdown(content)
+                        documents.append(html_content)  # Store HTML content or plain text
+                except Exception as e:
+                    print(f"Error reading {file_path}: {e}")
     return documents
 
 # Load documents at startup
@@ -62,7 +67,7 @@ def retrieve_relevant_context(query, documents):
     relevant_contexts = []
     
     for doc in documents:
-        if query.lower() in doc.lower():  # Basic keyword search (case insensitive)
+        if query.lower() in doc.lower():  # Basic keyword search
             relevant_contexts.append(doc)
 
     return relevant_contexts[:3]  # Return top 3 relevant contexts