Update app.py

app.py

@@ -103,7 +103,7 @@ def load_recommender(directory_path, start_page=1):
     recommender.fit(chunks)
     return 'Corpus Loaded.'
 
-def generate_text(openAI_key,prompt, engine="text-davinci-003"):
+def generate_text(openAI_key, prompt, engine="text-davinci-003"):
     openai.api_key = openAI_key
     completions = openai.Completion.create(
         engine=engine,
@@ -116,7 +116,7 @@ def generate_text(openAI_key,prompt, engine="text-davinci-003"):
     message = completions.choices[0].text
     return message
 
-def generate_answer(question,openAI_key):
+def generate_answer(question, openAI_key):
     topn_chunks = recommender(question)
     prompt = ""
     prompt += 'search results:\n\n'
@@ -133,7 +133,7 @@ def generate_answer(question,openAI_key):
               "answer should be short and concise. Answer step-by-step. \n\nQuery: {question}\nAnswer: "
     
     prompt += f"Query: {question}\nAnswer:"
-    answer = generate_text(openAI_key, prompt,"text-davinci-003")
+    answer = generate_text(openAI_key, prompt, "text-davinci-003")
     return answer
 
 
@@ -149,7 +149,8 @@ def question_answer(files, question, openAI_key):
         os.makedirs(directory_path)
 
     for file in files:
-        file.save(os.path.join(directory_path, file.name))
+        with open(os.path.join(directory_path, file.name), 'wb') as f:
+            f.write(file.read())
 
     load_recommender(directory_path)
 
@@ -159,6 +160,7 @@ def question_answer(files, question, openAI_key):
     return generate_answer(question, openAI_key)
 
 
+
 recommender = SemanticSearch()
 
 title = 'PDF GPT'
@@ -170,17 +172,19 @@ with gr.Blocks() as demo:
     gr.Markdown(description)
 
     with gr.Row():
-        
+
         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')
-            files = gr.Filebox(label='Upload PDF files', accept='.pdf', multiple=True)
-            question = gr.Textbox(label='Enter your question here')
+            openAI_key = gr.inputs.Textbox(label='Enter your OpenAI API key here')
+            files = gr.inputs.File(file_count="multiple", file_types=[".pdf"], label="Upload PDF files")
+            question = gr.inputs.Textbox(label='Enter your question here')
             btn = gr.Button(value='Submit')
 
         with gr.Group():
-            answer = gr.Textbox(label='The answer to your question is :')
+            answer = gr.outputs.Textbox(label='The answer to your question is :')
 
         btn.click(question_answer, inputs=[files, question, openAI_key], outputs=[answer])
-#openai.api_key = os.getenv('Your_Key_Here') 
-demo.launch()
+
+demo.launch()
+
+