Update app.py

app.py

@@ -83,7 +83,11 @@ if openai_api_key:
         "Code Generator", 
         "AI Chatbot Tutor", 
         "AI Study Notes & Summaries", 
-        "Code Bug Fixer"
+        "Code Bug Fixer",
+        "Mathematics Assistant",  # Added option for Math
+        "Biology Assistant",      # Added option for Biology
+        "Chemistry Assistant",    # Added option for Chemistry
+        "Physics Assistant"       # Added option for Physics
     ))
 
     if mode == "Course Query Assistant":
@@ -274,3 +278,86 @@ if openai_api_key:
                     )
             else:
                 st.error("Please enter some buggy code to fix.")
+
+    elif mode == "Mathematics Assistant":
+        st.header("Mathematics Assistant")
+
+        # Display image/logo in the "Mathematics Assistant" section (optional)
+        math_icon = Image.open("math_icon.png")  # Ensure the file is in the correct directory
+        st.image(math_icon, width=150)  # Adjust the size as per preference
+
+        # User input for math questions
+        math_query = st.text_input("Ask a mathematics-related question:")
+
+        if math_query:
+            with st.spinner("Getting answer..."):
+                prompt = f"Answer the following math question: {math_query}"
+                response = openai.ChatCompletion.create(
+                    model="gpt-4o-mini",
+                    messages=[{"role": "user", "content": prompt}]
+                )
+                answer = response['choices'][0]['message']['content']
+                st.write(f"### Answer: {answer}")
+
+    # **New Section: Biology Assistant**
+    elif mode == "Biology Assistant":
+        st.header("Biology Assistant")
+
+        # Display image/logo in the "Biology Assistant" section (optional)
+        bio_icon = Image.open("bio_icon.png")  # Ensure the file is in the correct directory
+        st.image(bio_icon, width=150)  # Adjust the size as per preference
+
+        # User input for biology questions
+        bio_query = st.text_input("Ask a biology-related question:")
+
+        if bio_query:
+            with st.spinner("Getting answer..."):
+                prompt = f"Answer the following biology question: {bio_query}"
+                response = openai.ChatCompletion.create(
+                    model="gpt-4o-mini",
+                    messages=[{"role": "user", "content": prompt}]
+                )
+                answer = response['choices'][0]['message']['content']
+                st.write(f"### Answer: {answer}")
+
+    # **New Section: Chemistry Assistant**
+    elif mode == "Chemistry Assistant":
+        st.header("Chemistry Assistant")
+
+        # Display image/logo in the "Chemistry Assistant" section (optional)
+        chem_icon = Image.open("chem_icon.png")  # Ensure the file is in the correct directory
+        st.image(chem_icon, width=150)  # Adjust the size as per preference
+
+        # User input for chemistry questions
+        chem_query = st.text_input("Ask a chemistry-related question:")
+
+        if chem_query:
+            with st.spinner("Getting answer..."):
+                prompt = f"Answer the following chemistry question: {chem_query}"
+                response = openai.ChatCompletion.create(
+                    model="gpt-4o-mini",
+                    messages=[{"role": "user", "content": prompt}]
+                )
+                answer = response['choices'][0]['message']['content']
+                st.write(f"### Answer: {answer}")
+
+    # **New Section: Physics Assistant**
+    elif mode == "Physics Assistant":
+        st.header("Physics Assistant")
+
+        # Display image/logo in the "Physics Assistant" section (optional)
+        phys_icon = Image.open("phys_icon.png")  # Ensure the file is in the correct directory
+        st.image(phys_icon, width=150)  # Adjust the size as per preference
+
+        # User input for physics questions
+        phys_query = st.text_input("Ask a physics-related question:")
+
+        if phys_query:
+            with st.spinner("Getting answer..."):
+                prompt = f"Answer the following physics question: {phys_query}"
+                response = openai.ChatCompletion.create(
+                    model="gpt-4o-mini",
+                    messages=[{"role": "user", "content": prompt}]
+                )
+                answer = response['choices'][0]['message']['content']
+                st.write(f"### Answer: {answer}")