Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import random
 
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+# Initialize conversation state
+conversation_state = {"ask_question": False, "last_message": ""}
 
 def respond(
     message,
@@ -15,6 +15,8 @@ def respond(
     temperature,
     top_p,
 ):
+    global conversation_state
+
     messages = [{"role": "system", "content": system_message}]
 
     for val in history:
@@ -25,6 +27,12 @@ def respond(
 
     messages.append({"role": "user", "content": message})
 
+    # Check if the chatbot should pose a question based on the user's previous response
+    if conversation_state["ask_question"]:
+        conversation_state["ask_question"] = False
+        question = pose_follow_up_question(conversation_state["last_message"])
+        messages.append({"role": "assistant", "content": question})
+
     response = ""
 
     for message in client.chat_completion(
@@ -37,11 +45,25 @@ def respond(
         token = message.choices[0].delta.content
 
         response += token
+
+        # Update conversation state with the last user message
+        conversation_state["last_message"] = message.choices[0].delta.content
+
         yield response
 
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
+def pose_follow_up_question(user_response):
+    # Example follow-up questions based on user responses
+    follow_up_questions = {
+        "I believe scientists should prioritize ethical considerations in their research":
+            "That's a commendable perspective! What measures do you think scientists should take to ensure ethical research practices?",
+        "I'm not sure about the ethical implications of genetic engineering":
+            "It's a complex topic indeed! What aspects of genetic engineering do you find most concerning?",
+        "I think technology has the potential to both benefit and harm society":
+            "You raise an interesting point! How do you think society can mitigate the risks associated with emerging technologies?"
+    }
+    return follow_up_questions.get(user_response, "I'd love to hear more about your thoughts on this topic.")
+
+# Gradio interface definition
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
@@ -58,6 +80,5 @@ demo = gr.ChatInterface(
     ],
 )
 
-
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()