Update app.py

app.py

@@ -110,9 +110,9 @@ def get_answer(query, history, temperature, top_p, max_token_count):
     Additionally, only if there is sufficient remaining token capacity, provide 2 or 3 related questions that the user might want to ask next based on the topic.
     If possible, use the following structure for these suggested questions:
     Here are some related questions you might be interested in:
-    <button style="background-color: #4CAF50; color: white; padding: 10px; margin: 5px; border: none; border-radius: 5px; cursor: pointer; font-size: 16px;">First related question</button>
-    <button style="background-color: #4CAF50; color: white; padding: 10px; margin: 5px; border: none; border-radius: 5px; cursor: pointer; font-size: 16px;">Second related question</button>
-    <button style="background-color: #4CAF50; color: white; padding: 10px; margin: 5px; border: none; border-radius: 5px; cursor: pointer; font-size: 16px;">Third related question</button>
+    <button onclick="document.getElementById('input_txt').value = this.innerHTML; document.getElementById('btn_submit').click();" style="background-color: #4CAF50; color: white; padding: 10px; margin: 5px; border: none; border-radius: 5px; cursor: pointer; font-size: 16px;">First related question</button>
+    <button onclick="document.getElementById('input_txt').value = this.innerHTML; document.getElementById('btn_submit').click();" style="background-color: #4CAF50; color: white; padding: 10px; margin: 5px; border: none; border-radius: 5px; cursor: pointer; font-size: 16px;">Second related question</button>
+    <button onclick="document.getElementById('input_txt').value = this.innerHTML; document.getElementById('btn_submit').click();" style="background-color: #4CAF50; color: white; padding: 10px; margin: 5px; border: none; border-radius: 5px; cursor: pointer; font-size: 16px;">Third related question</button>
     
     Ensure these suggested questions are brief, relevant, and encourage further exploration on the topic.
     Assistant:
@@ -186,7 +186,7 @@ def main():
     with gr.Blocks() as demo:
         chatbot = gr.Chatbot([[None, initial_message]], height=550)
 
-        msg = gr.Textbox(label="Question")
+        msg = gr.Textbox(label="Question", elem_id="input_txt")
         
         with gr.Accordion(label="Advanced options", open=False):
             temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1, value=0.6, step=0.1)
@@ -194,7 +194,7 @@ def main():
             max_token_count = gr.Slider(label="Max token count", minimum=1, maximum=1024, value=400, step=10)
             prompt_and_time = gr.Textbox(label="Prompt and Time", interactive=False)
         
-        btn = gr.Button("Submit")
+        btn = gr.Button("Submit", elem_id="btn_submit")
         clear = gr.Button("Clear console")
 
         btn.click(respond, inputs=[msg, chatbot, temperature, top_p, max_token_count], outputs=[msg, chatbot, prompt_and_time])