Update app.py

app.py

@@ -178,6 +178,13 @@ def display_response(response_key, response):
         file_name=f"{response_key}.txt"
     )
 
+# Function to handle image uploads
+def upload_image():
+    st.header("Image Upload")
+    image = st.file_uploader("Choose an image", type=["png", "jpg", "jpeg"])
+    if image is not None:
+        st.image(image, caption="Uploaded Image", use_column_width=True)
+
 if selected_tool == "Personalized Learning Assistant":
     st.header("Personalized Learning Assistant")
     with st.form(key="learning_form"):
@@ -188,17 +195,19 @@ if selected_tool == "Personalized Learning Assistant":
             st.session_state['responses']['personalized_learning_assistant'] = explanation
     if st.session_state['responses']['personalized_learning_assistant']:
         display_response("personalized_learning_assistant", st.session_state['responses']['personalized_learning_assistant'])
+    upload_image()
 
 elif selected_tool == "AI Coding Mentor":
     st.header("AI Coding Mentor")
     with st.form(key="coding_form"):
-        code_input = st.text_area("Enter your code snippet", placeholder="e.g., def add(a, b): return a + b")
-        submit_button = st.form_submit_button("Review Code")
+        code_input = st.text_area("Paste your code snippet", placeholder="e.g., for i in range(10): print(i)")
+        submit_button = st.form_submit_button("Get Review")
         if submit_button:
             review = ai_coding_mentor(code_input)
             st.session_state['responses']['ai_coding_mentor'] = review
     if st.session_state['responses']['ai_coding_mentor']:
         display_response("ai_coding_mentor", st.session_state['responses']['ai_coding_mentor'])
+    upload_image()
 
 elif selected_tool == "Smart Document Summarizer":
     st.header("Smart Document Summarizer")
@@ -210,6 +219,7 @@ elif selected_tool == "Smart Document Summarizer":
             st.session_state['responses']['smart_document_summarizer'] = summary
     if st.session_state['responses']['smart_document_summarizer']:
         display_response("smart_document_summarizer", st.session_state['responses']['smart_document_summarizer'])
+    upload_image()
 
 elif selected_tool == "Interactive Study Planner":
     st.header("Interactive Study Planner")
@@ -221,6 +231,7 @@ elif selected_tool == "Interactive Study Planner":
             st.session_state['responses']['interactive_study_planner'] = study_plan
     if st.session_state['responses']['interactive_study_planner']:
         display_response("interactive_study_planner", st.session_state['responses']['interactive_study_planner'])
+    upload_image()
 
 elif selected_tool == "Real-Time Q&A Support":
     st.header("Real-Time Q&A Support")
@@ -232,6 +243,7 @@ elif selected_tool == "Real-Time Q&A Support":
             st.session_state['responses']['real_time_qa_support'] = answer
     if st.session_state['responses']['real_time_qa_support']:
         display_response("real_time_qa_support", st.session_state['responses']['real_time_qa_support'])
+    upload_image()
 
 elif selected_tool == "Mental Health Check-In":
     st.header("Mental Health Check-In")
@@ -243,6 +255,7 @@ elif selected_tool == "Mental Health Check-In":
             st.session_state['responses']['mental_health_check_in'] = advice
     if st.session_state['responses']['mental_health_check_in']:
         display_response("mental_health_check_in", st.session_state['responses']['mental_health_check_in'])
+    upload_image()
 
 # Footer with social media links
 st.markdown("""