Update app.py

app.py

@@ -10,17 +10,30 @@ os.environ["GROQ_API_KEY"] = GROQ_API_KEY
 client = Groq(api_key=GROQ_API_KEY)
 
 # Streamlit user interface setup
-st.title("AI Chatbot")
-st.write("Hello! I'm your AI Study Assistant. Ask me anything, and I'll try to help.")
+st.title("Subject-specific AI Chatbot")
+st.write("Hello! I'm your AI Study Assistant. You can ask me any questions related to your subjects, and I'll try to help.")
 
 # Initialize session state for maintaining conversation
 if 'conversation_history' not in st.session_state:
     st.session_state.conversation_history = []
 
-# Function to generate chatbot response based on user input
+# Define a list of subjects for which the chatbot will answer
+subjects = ["Chemistry", "Computer", "English", "Islamiat", "Mathematics", "Physics", "Urdu"]
+
+# Function to generate chatbot response based on subject-specific user input
 def generate_chatbot_response(user_message):
-    prompt = f"You are a helpful AI chatbot. The user is asking: {user_message}. Provide a detailed, helpful response."
+    # Check if the user's question is related to any subject
+    related_subject = None
+    for subject in subjects:
+        if subject.lower() in user_message.lower():
+            related_subject = subject
+            break
     
+    if related_subject:
+        prompt = f"You are a helpful AI chatbot for studying {related_subject}. The user is asking: {user_message}. Provide a detailed, helpful response related to {related_subject}."
+    else:
+        prompt = f"You are a helpful AI chatbot. The user is asking: {user_message}. If the question is not related to any of the specified subjects (Chemistry, Computer, English, Islamiat, Mathematics, Physics, Urdu), politely let them know."
+
     # Generate response using Groq API
     chat_completion = client.chat.completions.create(
         messages=[{"role": "user", "content": prompt}],
@@ -31,7 +44,7 @@ def generate_chatbot_response(user_message):
     return response
 
 # User input for conversation
-user_input = st.text_input("Ask me anything:")
+user_input = st.text_input("Ask me a subject-related question:")
 
 # Handle user input and display conversation
 if user_input: