Update app.py

app.py

@@ -308,7 +308,7 @@ def summarization_interface():
             pdf_temperature = gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, value=0.7, step=0.01)
             pdf_top_p = gr.Slider(label="Top P", minimum=0.0, maximum=1.0, value=0.9, step=0.01)
             pdf_repetition_penalty = gr.Slider(label="Repetition Penalty", minimum=0.5, maximum=2.0, value=1.0, step=0.1)
-            pdf_summary_output = gr.Textbox(label="Summary Output")
+            pdf_summary_output = gr.Textbox(label="Concatenated Summary Output")
             pdf_summarize_button = gr.Button("Generate Summary")
             pdf_clear_cache_button = gr.Button("Clear Cache")
 
@@ -318,13 +318,21 @@ def summarization_interface():
             search_temperature = gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, value=0.7, step=0.01)
             search_top_p = gr.Slider(label="Top P", minimum=0.0, maximum=1.0, value=0.9, step=0.01)
             search_repetition_penalty = gr.Slider(label="Repetition Penalty", minimum=0.5, maximum=2.0, value=1.0, step=0.1)
-            search_summary_output = gr.Textbox(label="Summary Output")
+            search_summary_output = gr.Textbox(label="Concatenated Summary Output")
             search_summarize_button = gr.Button("Generate Summary")
             search_clear_cache_button = gr.Button("Clear Cache")
 
         # Bind functions to button clicks
-        pdf_summarize_button.click(fn=lambda file, instructions, temperature, top_p, repetition_penalty: process_input(file, True, instructions, temperature, top_p, repetition_penalty), inputs=[pdf_file, pdf_instructions, pdf_temperature, pdf_top_p, pdf_repetition_penalty], outputs=pdf_summary_output)
-        search_summarize_button.click(fn=lambda query, instructions, temperature, top_p, repetition_penalty: process_input(query, False, instructions, temperature, top_p, repetition_penalty), inputs=[search_query, search_instructions, search_temperature, search_top_p, search_repetition_penalty], outputs=search_summary_output)
+        pdf_summarize_button.click(
+            fn=lambda file, instructions, temperature, top_p, repetition_penalty: generate_and_save_summary(file, True, instructions, temperature, top_p, repetition_penalty),
+            inputs=[pdf_file, pdf_instructions, pdf_temperature, pdf_top_p, pdf_repetition_penalty],
+            outputs=[pdf_summary_output]
+        )
+        search_summarize_button.click(
+            fn=lambda query, instructions, temperature, top_p, repetition_penalty: generate_and_save_summary(query, False, instructions, temperature, top_p, repetition_penalty),
+            inputs=[search_query, search_instructions, search_temperature, search_top_p, search_repetition_penalty],
+            outputs=[search_summary_output]
+        )
         pdf_clear_cache_button.click(fn=clear_cache, inputs=None, outputs=pdf_summary_output)
         search_clear_cache_button.click(fn=clear_cache, inputs=None, outputs=search_summary_output)