split summary

app.py

@@ -37,20 +37,58 @@ def chat_with_pdf(question):
     prompt = ChatPromptTemplate.from_messages([
         ("human", pdf_content),
         ("human", question),
+        ("human", "Give a clear summary of this pdf information at a 8th grade reading level.")
     ])
 
     # Invoke the model using the chain
     chain = prompt | model
     response = chain.invoke({})
-    
-    return response.content
+
+    # Get the summary of the PDF content
+    summarizer = pipeline("summarization")
+    summary = summarizer(pdf_content, max_length=1000, min_length=30, do_sample=False)[0]['summary_text']
+
+    # Combine the chat response and the summary
+    combined_response = f"Summary: {summary}\n\nChat Response: {response.content}"
+
+    return combined_response
 
 # Define Gradio UI
+def gradio_interface(pdf_doc, question):
+    # ...
+    return gr.Interface(
+        fn=chat_with_pdf,
+        inputs=[pdf_doc, question],
+        outputs=gr.outputs.Textbox(),
+        api_name='chat_with_pdf_2'
+    )
+
 def gradio_interface(pdf_doc, question):
     if not pdf_content:
         return load_pdf(pdf_doc)
     else:
-        return chat_with_pdf(question)
+        # Get the summary of the PDF content
+        summarizer = pipeline("summarization")
+        summary = summarizer(pdf_content, max_length=100, min_length=30, do_sample=False)[0]['summary_text']
+
+        # Get the chat response
+        response = chat_with_pdf(question)
+
+        # Define the outputs
+        summary_output = gr.outputs.Textbox(label="Summary")
+        chat_output = gr.outputs.Textbox(label="Chat Response")
+
+        # Return the Gradio interface with the Multi output
+        return gr.Interface(
+            fn=chat_with_pdf,
+            inputs=[pdf_doc, question],
+            outputs=gradio.outputs.Multi(summary_output, chat_output),
+            examples=[["sample.pdf", "What is this document about?"]],
+            api_name='chat_with_pdf_2'
+        )
+
+gradio_interface(None, None)
+
 
 gr.Interface(fn=gradio_interface,
              inputs=[gr.File(label="Load a pdf", file_types=['.pdf'], type="file"),