Update app.py

app.py

@@ -156,6 +156,15 @@ selected_tool = st.sidebar.radio(
 # Display tool based on selection
 st.title("EduNexus :book:")
 
+# Common function to display response and download button
+def display_response(response_key, response):
+    st.write(response)
+    st.download_button(
+        "Download Response",
+        response,
+        file_name=f"{response_key}.txt"
+    )
+
 if selected_tool == "Personalized Learning Assistant":
     st.header("Personalized Learning Assistant")
     with st.form(key="learning_form"):
@@ -164,8 +173,8 @@ if selected_tool == "Personalized Learning Assistant":
         if submit_button:
             explanation = personalized_learning_assistant(topic_input)
             st.session_state['responses']['personalized_learning_assistant'] = explanation
-            st.write(explanation)
-            st.download_button("Download Response", explanation, file_name="explanation.txt")
+    if st.session_state['responses']['personalized_learning_assistant']:
+        display_response("personalized_learning_assistant", st.session_state['responses']['personalized_learning_assistant'])
 
 elif selected_tool == "AI Coding Mentor":
     st.header("AI Coding Mentor")
@@ -175,8 +184,8 @@ elif selected_tool == "AI Coding Mentor":
         if submit_button:
             review = ai_coding_mentor(code_input)
             st.session_state['responses']['ai_coding_mentor'] = review
-            st.write(review)
-            st.download_button("Download Response", review, file_name="review.txt")
+    if st.session_state['responses']['ai_coding_mentor']:
+        display_response("ai_coding_mentor", st.session_state['responses']['ai_coding_mentor'])
 
 elif selected_tool == "Smart Document Summarizer":
     st.header("Smart Document Summarizer")
@@ -186,8 +195,8 @@ elif selected_tool == "Smart Document Summarizer":
         if submit_button:
             summary = smart_document_summarizer(document_input)
             st.session_state['responses']['smart_document_summarizer'] = summary
-            st.write(summary)
-            st.download_button("Download Response", summary, file_name="summary.txt")
+    if st.session_state['responses']['smart_document_summarizer']:
+        display_response("smart_document_summarizer", st.session_state['responses']['smart_document_summarizer'])
 
 elif selected_tool == "Interactive Study Planner":
     st.header("Interactive Study Planner")
@@ -197,8 +206,8 @@ elif selected_tool == "Interactive Study Planner":
         if submit_button:
             plan = interactive_study_planner(exam_schedule_input)
             st.session_state['responses']['interactive_study_planner'] = plan
-            st.write(plan)
-            st.download_button("Download Response", plan, file_name="study_plan.txt")
+    if st.session_state['responses']['interactive_study_planner']:
+        display_response("interactive_study_planner", st.session_state['responses']['interactive_study_planner'])
 
 elif selected_tool == "Real-Time Q&A Support":
     st.header("Real-Time Q&A Support")
@@ -208,8 +217,8 @@ elif selected_tool == "Real-Time Q&A Support":
         if submit_button:
             answer = real_time_qa_support(question_input)
             st.session_state['responses']['real_time_qa_support'] = answer
-            st.write(answer)
-            st.download_button("Download Response", answer, file_name="answer.txt")
+    if st.session_state['responses']['real_time_qa_support']:
+        display_response("real_time_qa_support", st.session_state['responses']['real_time_qa_support'])
 
 elif selected_tool == "Mental Health Check-In":
     st.header("Mental Health Check-In")
@@ -219,12 +228,12 @@ elif selected_tool == "Mental Health Check-In":
         if submit_button:
             advice = mental_health_check_in(feelings_input)
             st.session_state['responses']['mental_health_check_in'] = advice
-            st.write(advice)
-            st.download_button("Download Response", advice, file_name="advice.txt")
+    if st.session_state['responses']['mental_health_check_in']:
+        display_response("mental_health_check_in", st.session_state['responses']['mental_health_check_in'])
 
 # Footer with acknowledgments
 st.markdown("""
-    <footer style="background-color: #1e1e2f; padding: 10px; text-align: center; color: #ffffff;">
+    <footer style="background-color: #1e1e2f; padding: 5px; text-align: center; color: #ffffff;">
         <p>&copy; 2024 EduNexus. All Rights Reserved.</p>
         <p>Created with ❤️ by [Your Name]</p>
         <p><a href="mailto:[email protected]" style="color: #ff9a8b;">Contact Us</a></p>