Update app.py

app.py

@@ -153,6 +153,10 @@ if openai_api_key:
     elif mode == "Code Generator":
         st.header("Code Generator")
 
+        # Display image/logo in the "Course Query Assistant" section (optional)
+        codegen = Image.open("9802381.png")  # Ensure the file is in the correct directory
+        st.image(codegen, width=150)  # Adjust the size as per preference
+
         # Code generation prompt input
         code_prompt = st.text_area("Describe the code you want to generate:", 
                                    "e.g., Write a Python program that generates Fibonacci numbers.")
@@ -189,6 +193,10 @@ if openai_api_key:
     elif mode == "AI Chatbot Tutor":
         st.header("AI Chatbot Tutor")
 
+        # Display image/logo in the "Course Query Assistant" section (optional)
+        aitut = Image.open("910372.png")  # Ensure the file is in the correct directory
+        st.image(aitut, width=150)  # Adjust the size as per preference
+
         # Chat interface for the AI tutor
         chat_history = []
 
@@ -233,6 +241,10 @@ if openai_api_key:
     elif mode == "Code Bug Fixer":
         st.header("Code Bug Fixer")
 
+        # Display image/logo in the "Course Query Assistant" section (optional)
+        aibug = Image.open("bug.png")  # Ensure the file is in the correct directory
+        st.image(aibug, width=150)  # Adjust the size as per preference
+
         # User input for buggy code
         buggy_code = st.text_area("Enter your buggy code here:")