Update app.py

app.py

@@ -256,7 +256,10 @@ def ocr_converter(input_file):
     ocrmypdf.ocr(image_pdf, image_pdf, skip_text=True, language="eng")
     ocrmypdf.ocr(image_pdf, image_pdf, redo_ocr=True, language="eng")
     return image_pdf
-    
+
+def summarize_contents():
+    question = "Generate a short summary of the contents along with no more than 3 leading/example questions"
+    return pdf_qa.run(question)    
 
 def answer_query(query):
     question = query
@@ -290,6 +293,10 @@ with gr.Blocks(css=css,theme=gr.themes.Monochrome()) as demo:
                 status = gr.Textbox(label="Status", placeholder="", interactive=False)
                 load_pdf = gr.Button("Upload PDF and generate embeddings").style(full_width=False)
 
+            with gr.Row():
+                summary = gr.Textbox(label="Summary")
+                summarize_pdf = gr.Button("Have Moli Summarize the Contents").style(full_width=False)
+
             with gr.Row():
                 input = gr.Textbox(label="Type in your question")
                 output = gr.Textbox(label="Answer")
@@ -581,6 +588,7 @@ with gr.Blocks(css=css,theme=gr.themes.Monochrome()) as demo:
         
     
     load_pdf.click(load_pdf_and_generate_embeddings, inputs=[pdf_doc, open_ai_key, relevant_pages], outputs=status)
+    summarize_pdf.click(summarize_contents,outputs=summary)
     load_csv.click(load_csv_and_store_questionset_into_sqlite, inputs=[csv_file, document_type_for_questionset, tag_for_questionset], outputs=status_for_loading_csv)
 
     load_questionsets.click(retrieve_document_type_and_questionsettag_from_sqlite,outputs=questionsets)