Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import gradio as gr
+import asyncio
 from langchain import PromptTemplate
 from langchain.chains.question_answering import load_qa_chain
 from langchain.document_loaders import PyPDFLoader
@@ -7,9 +8,8 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 import google.generativeai as genai
 
 
-
 # Fungsi untuk inisialisasi
-def initialize(file_path, question):
+async def initialize(file_path, question):
     genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
     model = genai.GenerativeModel('gemini-pro')
     model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3)
@@ -25,20 +25,21 @@ def initialize(file_path, question):
         pages = pdf_loader.load_and_split()
         context = "\n".join(str(page.page_content) for page in pages[:30])
         stuff_chain = load_qa_chain(model, chain_type="stuff", prompt=prompt)
-        stuff_answer = stuff_chain({"input_documents": pages, "question": question, "context": context}, return_only_outputs=True)
+        stuff_answer = await stuff_chain({"input_documents": pages, "question": question, "context": context}, return_only_outputs=True)
         return stuff_answer['output_text']
     else:
         return "Error: Unable to process the document. Please ensure the PDF file is valid."
 
 # Define Gradio Interface
-input_file = gr.File(label="Upload PDF File")
-input_question = gr.Textbox(label="Ask about the document")
-output_text = gr.Textbox(label="Answer - GeminiPro")
+input_file = gr.inputs.File(label="Upload PDF File")
+input_question = gr.inputs.Textbox(label="Ask about the document")
+output_text = gr.outputs.Textbox(label="Answer - GeminiPro")
 
-def pdf_qa(file, question):
-    answer = initialize(file.name, question)
+async def pdf_qa(file, question):
+    answer = await initialize(file.name, question)
     return answer
 
 # Create Gradio Interface
 gr.Interface(fn=pdf_qa, inputs=[input_file, input_question], outputs=output_text, title="PDF Question Answering System", description="Upload a PDF file and ask questions about the content.").launch()
 
+