Update main.py

main.py

@@ -9,9 +9,20 @@ model = LlamaForCausalLM.from_pretrained(model_name)
 # Initialize the text generation pipeline
 llm_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# Load the prompt from the text file
-with open('website_text.txt', 'r') as file:
-    prompt = file.read()
+import fitz  # PyMuPDF
+
+# Function to extract text from PDF
+def extract_text_from_pdf(pdf_path):
+    document = fitz.open(pdf_path)
+    text = ""
+    for page_num in range(document.page_count):
+        page = document.load_page(page_num)
+        text += page.get_text()
+    return text
+
+# Load the prompt from the PDF file
+pdf_path = 'landon_Hotel.pdf'
+prompt = extract_text_from_pdf(pdf_path)