Update app.py

app.py

@@ -5,7 +5,7 @@ from langchain_openai import ChatOpenAI
 from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessagePromptTemplate, AIMessagePromptTemplate
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnablePassthrough
-from langchain_community.document_loaders import TextLoader, UnstructuredWordDocumentLoader
+from langchain_community.document_loaders import TextLoader
 
 # Configuration
 SECRET_KEY = "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"  # Replace with your actual API key
@@ -23,14 +23,28 @@ def load_runbooks():
     runbooks = {}
     for file in os.listdir(RUNBOOK_DIR):
         path = os.path.join(RUNBOOK_DIR, file)
-        if file.endswith(".txt"):
-            loader = TextLoader(path)
-        elif file.endswith(".docx"):
-            loader = UnstructuredWordDocumentLoader(path)
-        else:
-            continue
-        docs = loader.load()
-        runbooks[file] = "\n".join([doc.page_content for doc in docs])
+        try:
+            if file.endswith(".txt"):
+                # Load text files using TextLoader
+                loader = TextLoader(path)
+                docs = loader.load()
+                content = "\n".join([doc.page_content for doc in docs])
+            
+            elif file.endswith(".docx"):
+                # Load .docx files using python-docx
+                doc = DocxDocument(path)
+                content = "\n".join([para.text for para in doc.paragraphs])
+            
+            else:
+                # Skip unsupported file types
+                continue
+            
+            # Add the file's content to the runbooks dictionary
+            runbooks[file] = content
+        
+        except Exception as e:
+            print(f"Error loading file {file}: {e}")
+    
     return runbooks
 
 RUNBOOKS = load_runbooks()