Update app.py

app.py

@@ -29,7 +29,7 @@ if not hf_token:
 
 # Constants
 MODEL_NAME = "meta-llama/Llama-2-7b-chat-hf"
-KNOWLEDGE_BASE_DIR = "knowledge_base"
+KNOWLEDGE_BASE_DIR = "."  # Changed to root directory
 
 class DocumentLoader:
     """Class to manage PDF document loading."""
@@ -37,7 +37,11 @@ class DocumentLoader:
     @staticmethod
     def load_pdfs(directory_path: str) -> List:
         documents = []
-        pdf_files = [f for f in os.listdir(directory_path) if f.endswith('.pdf')]
+        pdf_files = [
+            f for f in os.listdir(directory_path) 
+            if f.endswith('.pdf') and 
+            (f.startswith('valencia') or 'fislac' in f.lower() or 'Valencia' in f)
+        ]
         
         for pdf_file in pdf_files:
             pdf_path = os.path.join(directory_path, pdf_file)
@@ -48,7 +52,7 @@ class DocumentLoader:
                 for doc in pdf_documents:
                     doc.metadata.update({
                         'title': pdf_file,
-                        'type': 'technical' if 'Valencia' in pdf_file else 'qa',
+                        'type': 'technical' if 'valencia' in pdf_file.lower() or 'Valencia' in pdf_file else 'qa',
                         'language': 'en',
                         'page': doc.metadata.get('page', 0)
                     })
@@ -58,6 +62,9 @@ class DocumentLoader:
             except Exception as e:
                 logger.error(f"Error loading {pdf_file}: {str(e)}")
         
+        if not documents:
+            logger.warning("No PDF documents found in the specified directory")
+            
         return documents
 
 class TextProcessor:
@@ -334,4 +341,4 @@ try:
 
 except Exception as e:
     logger.error(f"Error in Gradio interface creation: {str(e)}")
-    raise
+    raise