Update app.py

app.py

@@ -15,6 +15,7 @@ from langchain_core.documents import Document
 from huggingface_hub import InferenceClient
 import inspect
 import logging
+from gradio.exceptions import CancelledError
 
 
 # Set up basic configuration for logging
@@ -244,7 +245,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, use_web_search, model, temperature, num_calls, selected_docs, continuation=False):
+def respond(message, history, model, temperature, num_calls, use_web_search, selected_docs):
     if not message.strip():
         return "", history
 
@@ -255,17 +256,17 @@ def respond(message, history, use_web_search, model, temperature, num_calls, sel
             for main_content, sources in get_response_with_search(message, model, num_calls=num_calls, temperature=temperature):
                 response = f"{main_content}\n\n{sources}"
                 history[-1] = (message, response)
-                yield history, gr.update(visible=True)  # Make Continue Generation button visible
+                yield history
         else:
-            for partial_response in get_response_from_pdf(message, model, selected_docs, num_calls=num_calls, temperature=temperature, continuation=continuation):
+            for partial_response in get_response_from_pdf(message, model, selected_docs, num_calls=num_calls, temperature=temperature):
                 history[-1] = (message, partial_response)
-                yield history, gr.update(visible=True)  # Make Continue Generation button visible
-    except gr.CancelledError:
-        yield history, gr.update(visible=False)
+                yield history
+    except CancelledError:
+        yield history
     except Exception as e:
         logging.error(f"Unexpected error in respond: {str(e)}")
         history[-1] = (message, f"An unexpected error occurred: {str(e)}")
-        yield history, gr.update(visible=False)
+        yield history
 
 logging.basicConfig(level=logging.DEBUG)
 
@@ -526,12 +527,7 @@ with demo:
     
     # Add the click event for the Continue Generation button
     continue_btn.click(continue_generation, 
-                       inputs=[demo.chatbot, 
-                               gr.Dropdown(choices=MODELS, label="Select Model"), 
-                               gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature"),
-                               gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls"),
-                               use_web_search, 
-                               document_selector], 
+                       inputs=[demo.chatbot] + demo.additional_inputs,  # Use existing inputs
                        outputs=[demo.chatbot])
 
     gr.Markdown(