Update app.py

app.py

@@ -39,6 +39,18 @@ def generate_response(selected_question):
             temperature=0.7,
         ).choices[0].message['content'].strip()
         
+        follow_up_prompt = f"Based on the following response, suggest three follow-up questions: {response}"
+        follow_up_response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "system", "content": "You are a friendly and helpful chatbot."},
+                {"role": "user", "content": follow_up_prompt}
+            ],
+            max_tokens=100,
+            temperature=0.7,
+        ).choices[0].message['content'].strip()
+
+        follow_up_questions = follow_up_response.split('\n')
         topics_str = "Topic analysis not available"
         
         # Adjusted to return a list of tuples as expected by the Chatbot component
@@ -47,8 +59,12 @@ def generate_response(selected_question):
     except Exception as e:
         new_response = (None, f"Error generating response: {e}")
         chat_history.append(new_response)
+        follow_up_questions = []
     
-    return chat_history
+    return chat_history, follow_up_questions
+
+def update_examples(follow_up_questions):
+    return gr.Examples.update(examples=[[q] for q in follow_up_questions])
 
 with gr.Blocks() as demo:
     gr.Markdown(
@@ -70,16 +86,20 @@ with gr.Blocks() as demo:
         txt = gr.Textbox(scale=4, show_label=False, placeholder="Select Question", container=False, interactive=False)  # Adjust based on need
         btn = gr.Button("Submit")
 
-    btn.click(fn=generate_response, inputs=[txt], outputs=chatbot)
-
-    examples = [
-        ["What are the basic requirements to become an airforce pilot?"],
-        ["How long does it take to train as an airforce pilot?"],
-        ["Can you describe a day in the life of an airforce pilot?"]
-    ]
-    gr.Examples(examples, inputs=[txt], outputs=[chatbot], label="Select Question")
+    btn.click(fn=generate_response, inputs=[txt], outputs=[chatbot, 'examples'])
+    
+    examples_component = gr.Examples(
+        examples=[
+            ["What are the basic requirements to become an airforce pilot?"],
+            ["How long does it take to train as an airforce pilot?"],
+            ["Can you describe a day in the life of an airforce pilot?"]
+        ],
+        inputs=[txt],
+        outputs=[chatbot],
+        label="Select Question"
+    )
 
     chatbot.like(print_like_dislike, None, None)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()