Update app.py

app.py

@@ -7,6 +7,10 @@ import openai
 import gradio as gr
 import os
 from sklearn.neighbors import NearestNeighbors
+from datasets import load_dataset
+
+dataset = load_dataset('text', data_files='path_to_your_directory/*.txt')
+
 
 def download_pdf(url, output_path):
     urllib.request.urlretrieve(url, output_path)
@@ -131,6 +135,7 @@ def generate_answer(question,openAI_key):
     return answer
 
 
+'''
 def question_answer(url, file, question,openAI_key):
     if openAI_key.strip()=='':
         return '[ERROR]: Please enter you Open AI Key. Get your key here : https://platform.openai.com/account/api-keys'
@@ -156,6 +161,22 @@ def question_answer(url, file, question,openAI_key):
         return '[ERROR]: Question field is empty'
 
     return generate_answer(question,openAI_key)
+'''
+
+# New code for single staticPDF
+def question_answer(question, openAI_key):
+  # Check for empty OpenAI key or question
+  if openAI_key.strip() == '':
+    return '[ERROR]: Please enter you Open AI Key. Get your key here : https://platform.openai.com/account/api-keys'
+  if question.strip() == '':
+    return '[ERROR]: Question field is empty'
+
+  # URL of your PDF
+  url = 'https://huggingface.co/datasets/maconphillips/Warren-VT-Info/resolve/main/town-plan.pdf'
+  download_pdf(url, 'corpus.pdf')  
+  load_recommender('corpus.pdf')  
+  
+  return generate_answer(question, openAI_key)
 
 
 recommender = SemanticSearch()