Update app.py

app.py

@@ -64,6 +64,14 @@ client = Groq(api_key="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"
 
 
 
+
+
+
+
+
+
+import streamlit as st
+
 # Sidebar Menu with topics and subtopics
 topics = {
     "Artificial Intelligence": ["Overview", "Applications", "Future Prospects"],
@@ -84,6 +92,9 @@ for topic, subtopics in topics.items():
         if selected_subtopic:
             selected_topic = topic
 
+# Checkbox to confirm user is ready to view response
+show_response = st.sidebar.checkbox("Show response")
+
 # Display title based on subtopic selection
 if selected_subtopic:
     st.title(f"MEGA_BOT - {selected_subtopic}")
@@ -99,101 +110,89 @@ def generate_response(prompt):
     )
     return chat_completion.choices[0].message.content
 
-# Display response based on the selected subtopic
-if selected_subtopic and selected_topic:
-    # Creating a prompt for the model based on the selected subtopic
-    if selected_subtopic == "Overview" and selected_topic == "Artificial Intelligence":
-        prompt = """
+# Mapping of topic-subtopic pairs to prompts
+prompts = {
+    ("Artificial Intelligence", "Overview"): """
         Explain the overview of Artificial Intelligence (AI) in a detailed manner.
         Break down the explanation into simple, easy-to-understand points for beginners, avoiding technical jargon.
         Make it long and creative, and include practical examples and analogies where possible.
-        """
-
-    elif selected_subtopic == "Applications" and selected_topic == "Artificial Intelligence":
-        prompt = """
+    """,
+    ("Artificial Intelligence", "Applications"): """
         Explain the main applications of Artificial Intelligence in various industries.
         Break it down into point-wise, simple, and creative explanations suitable for a beginner.
         Include examples from everyday life to make it relatable.
-        """
-
-    elif selected_subtopic == "Future Prospects" and selected_topic == "Artificial Intelligence":
-        prompt = """
+    """,
+    ("Artificial Intelligence", "Future Prospects"): """
         What are the future prospects of Artificial Intelligence, and how will it evolve?
         Provide a detailed response with creative ideas, breaking it down into simple, easy-to-understand points.
         Consider the potential positive and negative impacts on society.
-        """
-
-    elif selected_subtopic == "Introduction" and selected_topic == "Generative AI":
-        prompt = """
+    """,
+    ("Generative AI", "Introduction"): """
         Provide an introduction to Generative AI and its core principles.
         Use simple, beginner-friendly language and provide creative explanations with examples.
         Break it down into points and keep the response long and informative.
-        """
-
-    elif selected_subtopic == "Applications" and selected_topic == "Generative AI":
-        prompt = """
+    """,
+    ("Generative AI", "Applications"): """
         Explain the key applications of Generative AI in various fields.
         Break the explanation into simple, easy-to-understand points, and use examples.
         Make the response long and creative with real-world use cases.
-        """
-
-    elif selected_subtopic == "Ethics" and selected_topic == "Generative AI":
-        prompt = """
+    """,
+    ("Generative AI", "Ethics"): """
         What are the ethical concerns related to Generative AI and its usage?
         Provide a detailed, easy-to-understand explanation with examples.
         Break down the explanation into simple, point-wise sections suitable for a beginner.
-        """
-
-    elif selected_subtopic == "GPT" and selected_topic == "Specific Models":
-        prompt = """
+    """,
+    ("Specific Models", "GPT"): """
         Explain the GPT (Generative Pretrained Transformer) model and its capabilities.
         Break the explanation down into beginner-friendly, simple points.
         Use analogies and relatable examples to make it creative and easy to understand.
-        """
-
-    elif selected_subtopic == "BERT" and selected_topic == "Specific Models":
-        prompt = """
+    """,
+    ("Specific Models", "BERT"): """
         Explain the BERT (Bidirectional Encoder Representations from Transformers) model.
         Provide a detailed, easy-to-understand response with simple points and examples.
         Use creative language to explain complex concepts simply.
-        """
-
-    elif selected_subtopic == "Electra" and selected_topic == "Specific Models":
-        prompt = """
+    """,
+    ("Specific Models", "Electra"): """
         Describe the Electra model and its applications.
         Break the explanation into simple, easy-to-understand points, and keep it creative and beginner-friendly.
         Provide relatable examples where possible.
-        """
-
-    elif selected_subtopic == "Gemma2-9b-it" and selected_topic == "Specific Models":
-        prompt = """
+    """,
+    ("Specific Models", "Gemma2-9b-it"): """
         What is the Gemma2-9b-it model, and what makes it unique?
         Break down the explanation into points and make it understandable for a beginner.
         Include simple examples and analogies to help explain the key features and capabilities.
-        """
-
-    elif selected_subtopic == "With RAG" and selected_topic == "Building Applications":
-        prompt = """
+    """,
+    ("Building Applications", "With RAG"): """
         Explain how to build applications using Retrieval-Augmented Generation (RAG).
         Break it down into beginner-friendly points with creative examples and analogies.
         Provide a detailed yet simple explanation for non-technical users.
-        """
-
-    elif selected_subtopic == "Without RAG" and selected_topic == "Building Applications":
-        prompt = """
+    """,
+    ("Building Applications", "Without RAG"): """
         Explain how to build applications without using Retrieval-Augmented Generation (RAG).
         Provide a detailed, easy-to-understand explanation with creative examples.
         Break down the response into simple, point-wise sections for beginners.
-        """
+    """
+}
 
+# Display response based on the selected subtopic only if the checkbox is checked
+if selected_subtopic and selected_topic and show_response:
+    # Retrieve the prompt based on the selected topic and subtopic
+    prompt = prompts.get((selected_topic, selected_subtopic), "No prompt available for this selection.")
+    
     # Generate the response based on the prompt using the Groq model
     response = generate_response(prompt)
-
+    
     # Display the model's response
     st.write(response)
 
-else:
-    st.write("Please select a subtopic to learn more.")
+elif not show_response:
+    # Display a message prompting the user to check the box if it is not checked
+    st.write("Please select a topic and subtopic from the sidebar and check 'Show response' to see more information.")
+
+
+
+
+