Update app.py

app.py

@@ -1,3 +1,4 @@
+import glob
 import urllib.request
 import fitz
 import re
@@ -136,12 +137,19 @@ def generate_answer(question,openAI_key):
     return answer
 
 
-def question_answer(directory_path, question, openAI_key):
+def question_answer(files, question, openAI_key):
     if openAI_key.strip() == '':
         return '[ERROR]: Please enter your Open AI Key. Get your key here : https://platform.openai.com/account/api-keys'
 
-    if not os.path.isdir(directory_path):
-        return '[ERROR]: Invalid directory path.'
+    if len(files) == 0:
+        return '[ERROR]: No PDF files uploaded.'
+
+    directory_path = os.path.join(os.getcwd(), 'uploaded_files')
+    if not os.path.exists(directory_path):
+        os.makedirs(directory_path)
+
+    for file in files:
+        file.save(os.path.join(directory_path, file.name))
 
     load_recommender(directory_path)
 
@@ -166,15 +174,13 @@ with gr.Blocks() as demo:
         with gr.Group():
             gr.Markdown(f'<p style="text-align:center">Get your Open AI API key <a href="https://platform.openai.com/account/api-keys">here</a></p>')
             openAI_key = gr.Textbox(label='Enter your OpenAI API key here')
-            directory_path = gr.Textbox(label='Enter the directory path containing PDF files')
+            files = gr.Filebox(label='Upload PDF files', accept='.pdf', multiple=True)
             question = gr.Textbox(label='Enter your question here')
             btn = gr.Button(value='Submit')
 
         with gr.Group():
             answer = gr.Textbox(label='The answer to your question is :')
 
-        #btn.click(question_answer, inputs=[url, file, question,openAI_key], outputs=[answer])
-        btn.click(question_answer, inputs=[directory_path, question, openAI_key], outputs=[answer])
-
+        btn.click(question_answer, inputs=[files, question, openAI_key], outputs=[answer])
 #openai.api_key = os.getenv('Your_Key_Here') 
 demo.launch()