Update app.py

app.py

@@ -65,6 +65,15 @@ rag_chain = (
     | StrOutputParser()
 )
 
+# Generate follow-up questions dynamically
+def generate_follow_ups(response):
+    # Simple logic to generate follow-up questions dynamically
+    # You can replace this with a more sophisticated model-based approach
+    return [
+        f"What more can you tell me about {response.split()[0]}?",
+        f"Can you elaborate on {response.split()[-1]}?",
+    ]
+
 
 
 # Function to handle chatbot interaction
@@ -72,13 +81,30 @@ def rag_chain_response(messages, user_message):
     # Generate a response using the RAG chain
     response = rag_chain.invoke(user_message)
     
+    # Generate follow-up questions based on the response
+    follow_ups = generate_follow_ups(response)
+    
     # Append the user's message and the response to the chat
     messages.append((user_message, response))
     
-    # Return the updated chat and clear the input box
-    return messages, ""
+    # Return the updated chat, follow-up questions, and clear the input box
+    return messages, follow_ups, ""
 
+# Function to handle follow-up clicks
+def follow_up_click(messages, follow_up_question):
+    # Treat the follow-up question as the user query
+    response = rag_chain.invoke(follow_up_question)
+    
+    # Generate new follow-ups based on the new response
+    follow_ups = generate_follow_ups(response)
+    
+    # Append the follow-up question and response to the chat
+    messages.append((follow_up_question, response))
+    
+    # Return the updated chat and follow-up questions
+    return messages, follow_ups
 
+    
 
 # Define the Gradio app
 with gr.Blocks(theme="rawrsor1/Everforest") as app:
@@ -87,17 +113,25 @@ with gr.Blocks(theme="rawrsor1/Everforest") as app:
     chatbot = gr.Chatbot([], elem_id="RADAR", bubble_full_width=False)
     question_input = gr.Textbox(label="Ask a Question", placeholder="Type your question here...")
     submit_btn = gr.Button("Submit")
+    follow_up_btns = gr.ButtonGroup(label="Follow-Up Questions", buttons=[""] * 2)
     
     # Set up interaction for both Enter key and Submit button
     question_input.submit(
         rag_chain_response,          # Function to handle input and generate response
         inputs=[chatbot, question_input],  # Pass current conversation state and user input
-        outputs=[chatbot, question_input]  # Update conversation state and clear the input
+        outputs=[chatbot, follow_up_btns, question_input]  # Update chat, follow-ups, and clear input
     )
     submit_btn.click(
         rag_chain_response,          # Function to handle input and generate response
         inputs=[chatbot, question_input],  # Pass current conversation state and user input
-        outputs=[chatbot, question_input]  # Update conversation state and clear the input
+        outputs=[chatbot, follow_up_btns, question_input]  # Update chat, follow-ups, and clear input
+    )
+    
+    # Handle follow-up button clicks
+    follow_up_btns.click(
+        follow_up_click,
+        inputs=[chatbot, follow_up_btns],
+        outputs=[chatbot, follow_up_btns]
     )
 
 # Launch the Gradio app