Update app.py

app.py

@@ -243,7 +243,7 @@ def retry_last_response(history, use_web_search, model, temperature, num_calls):
     
     return chatbot_interface(last_user_msg, history, use_web_search, model, temperature, num_calls)
 
-def respond(message, history, model, temperature, num_calls, use_web_search, selected_docs, instruction_key=None):
+def respond(message, history, model, temperature, num_calls, use_web_search, selected_docs, instruction_key):
     logging.info(f"User Query: {message}")
     logging.info(f"Model Used: {model}")
     logging.info(f"Search Type: {'Web Search' if use_web_search else 'PDF Search'}")
@@ -251,7 +251,7 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
     logging.info(f"Instruction Key: {instruction_key}")
 
     try:
-        if instruction_key:
+        if instruction_key and instruction_key != "None":
             # This is a summary generation request
             instruction = INSTRUCTION_PROMPTS[instruction_key]
             context_str = get_context_for_summary(selected_docs)
@@ -535,12 +535,7 @@ use_web_search = gr.Checkbox(label="Use Web Search", value=True)
 
 custom_placeholder = "Ask a question (Note: You can toggle between Web Search and PDF Chat in Additional Inputs below)"
 
-instruction_dropdown = gr.Dropdown(choices=list(INSTRUCTION_PROMPTS.keys()), label="Select Entity Type", value=list(INSTRUCTION_PROMPTS.keys())[0])
-summary_button = gr.Button("Generate Summary")
-
-def generate_summary(instruction_key, history, model, temperature, num_calls, use_web_search, selected_docs):
-    summary_response = next(respond("", history, model, temperature, num_calls, use_web_search, selected_docs, instruction_key))
-    return history + [("Generate summary for " + instruction_key, summary_response)]
+instruction_choices = ["None"] + list(INSTRUCTION_PROMPTS.keys())
 
 demo = gr.ChatInterface(
     respond,
@@ -550,10 +545,10 @@ demo = gr.ChatInterface(
         gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls"),
         use_web_search,
         document_selector,
-        instruction_dropdown,
+        gr.Dropdown(choices=instruction_choices, label="Select Entity Type for Summary", value="None")
     ],
     title="AI-powered Web Search and PDF Chat Assistant",
-    description="Chat with your PDFs or use web search to answer questions. Toggle between Web Search and PDF Chat in Additional Inputs below.",
+    description="Chat with your PDFs, use web search to answer questions, or generate summaries. Select an Entity Type for Summary to generate a specific summary.",
     theme=gr.themes.Soft(
         primary_hue="orange",
         secondary_hue="amber",
@@ -589,12 +584,6 @@ demo = gr.ChatInterface(
     )
 )
 
-summary_button.click(
-    generate_summary,
-    inputs=[instruction_dropdown, demo.chatbot, demo.additional_inputs[0], demo.additional_inputs[1], demo.additional_inputs[2], demo.additional_inputs[3], demo.additional_inputs[4]],
-    outputs=[demo.chatbot]
-)
-
 # Add file upload functionality
 with demo:
     gr.Markdown("## Upload PDF Documents")