Update app.py

app.py

@@ -86,31 +86,27 @@ def add_html_to_word(html_text, doc):
         elif element.name:
             doc.add_paragraph(element.get_text())  # For any other tags
 
-def process_pdfs(pdf_files, user_questions):
-    combined_report_text = ""
+def process_pdf(pdf_file, user_questions):
+    file_name = pdf_file.split('/')[-1]
+    saved_file_path = f"/tmp/{file_name}"
+    shutil.copyfile(pdf_file, saved_file_path)
 
-    for pdf_file in pdf_files:
-        file_name = pdf_file.orig_name
-        saved_file_path = f"/tmp/{file_name}"
-        shutil.copyfile(pdf_file.name, saved_file_path)
+    subprocess.run(["apt-get", "update"])
+    subprocess.run(["apt-get", "install", "-y", "poppler-utils"])
+    subprocess.run(["pdftotext", saved_file_path, "/tmp/text_file.txt"])
 
-        subprocess.run(["apt-get", "update"])
-        subprocess.run(["apt-get", "install", "-y", "poppler-utils"])
-        subprocess.run(["pdftotext", saved_file_path, "/tmp/text_file.txt"])
+    text_file = upload_file("/tmp/text_file.txt")
+    chat_session = build_model(text_file)
 
-        text_file = upload_file("/tmp/text_file.txt")
-        chat_session = build_model(text_file)
-
-        questions = user_questions.strip().split('\n')
-        report_text = generate_report(chat_session, questions)
-
-        combined_report_text += report_text + "\n\n"
+    questions = user_questions.strip().split('\n')
+    report_text = generate_report(chat_session, questions)
 
     doc = Document()
-    html_text = convert_markdown_to_html(combined_report_text)
+    html_text = convert_markdown_to_html(report_text)
     add_html_to_word(html_text, doc)
 
-    doc_name = "Report_combined.docx"
+    doc_name = file_name.replace(".pdf", ".docx")
+    doc_name = "Report_" + doc_name
     doc.save(f"/tmp/{doc_name}")
 
     return html_text, f"/tmp/{doc_name}"
@@ -125,18 +121,18 @@ questions = [
 questions_str = "\n".join(questions)
 
 iface = gr.Interface(
-    fn=process_pdfs,
+    fn=process_pdf,
     inputs=[
-        gr.Files(label="Upload PDF(s)", type="filepath"),
+        gr.File(label="Upload PDF", type="filepath"),
         gr.TextArea(label="Enter Questions", placeholder="Type your questions here, one per line.", value=questions_str)
     ],
     outputs=[
         gr.HTML(label="HTML Formatted Report"),
         gr.File(label="DOCX File Output", type="binary")
     ],
-    title="REPORT GENERATOR: ASK YOUR QUESTIONS TO PDF FILES @YED",
-    description="Upload multiple PDFs to ask questions and get the answers."
+    title="REPORT GENERATOR: ASK YOUR QUESTIONS TO A PDF FILE BY @YED",
+    description="Upload a PDF to ask questions and get the answers."
 )
 
 setup_api_key()
-iface.launch(share=True)
+iface.launch()