Update app.py

app.py

@@ -65,14 +65,8 @@ 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]}?",
-    ]
+
+
 
 
 
@@ -81,28 +75,13 @@ 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, follow-up questions, and clear the input box
-    return messages, follow_ups, ""
+    # Return the updated chat and clear the input box
+    return messages, ""
+
 
-# 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
 
     
 
@@ -113,25 +92,17 @@ 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, follow_up_btns, question_input]  # Update chat, follow-ups, and clear input
+        outputs=[chatbot, question_input]  # Update conversation state and clear the 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, 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]
+        outputs=[chatbot, question_input]  # Update conversation state and clear the input
     )
 
 # Launch the Gradio app