Update app.py

app.py

@@ -6,7 +6,7 @@ from groq import Groq
 st.set_page_config(page_title="EduNexus", page_icon=":book:", layout="wide")
 
 # Set the Groq API key
-os.environ["GROQ_API_KEY"] = "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"  # Replace with your actual API key
+os.environ["GROQ_API_KEY"] = "your_groq_api_key_here"  # Replace with your actual API key
 
 # Initialize Groq client
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
@@ -25,7 +25,7 @@ def call_groq_api(prompt):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Define functions for each tool with few-shot examples
+# Define functions for each tool with examples
 def personalized_learning_assistant(topic):
     prompt = f"""
     Provide a personalized learning plan for the topic: {topic}.
@@ -98,8 +98,9 @@ if 'responses' not in st.session_state:
 
 # Function to clear session state values
 def clear_session_state():
-    for key in st.session_state['responses'].keys():
-        st.session_state['responses'][key] = ""
+    for key in st.session_state.keys():
+        if key in ['personalized_learning_assistant', 'ai_coding_mentor', 'smart_document_summarizer', 'interactive_study_planner', 'real_time_qa_support', 'mental_health_check_in']:
+            st.session_state[key] = ""
 
 # Add custom styling using Streamlit
 st.markdown("""
@@ -108,10 +109,6 @@ st.markdown("""
         background-color: #000000; /* Royal Black */
         color: #ffffff; /* White */
     }
-    .css-1o7k8tt {
-        background-color: #000000; /* Royal Black */
-        color: #ffffff; /* White */
-    }
     .stButton {
         background-color: #ffffff; /* White */
         color: #000000; /* Royal Black */
@@ -135,16 +132,6 @@ st.markdown("""
         color: #000000; /* Royal Black */
         font-weight: bold;
     }
-    .stMarkdown, .stTextInput, .stTextArea {
-        border-radius: 12px;
-    }
-    .stSidebar {
-        background-color: #000000; /* Royal Black */
-        color: #ffffff; /* White */
-    }
-    .stSidebar .stMarkdown {
-        color: #ffffff; /* White */
-    }
     .footer {
         background-color: #000000; /* Royal Black */
         padding: 15px; 
@@ -167,11 +154,6 @@ st.markdown("""
     .footer i {
         font-size: 22px;
     }
-    .css-1n8h7zm .stTextArea::placeholder, 
-    .css-1n8h7zm .stTextInput::placeholder {
-        color: #000000; /* Royal Black */
-        font-weight: bold;
-    }
     </style>
 """, unsafe_allow_html=True)
 
@@ -185,63 +167,97 @@ tasks = [
     "🤖 AI Coding Mentor",
     "📄 Smart Document Summarizer",
     "🗓️ Interactive Study Planner",
-    "❓ Real-Time Q&A Support",
-    "💬 Mental Health Check-In"
+    "💬 Real-Time Q&A Support",
+    "🧠 Mental Health Check-In"
 ]
+
 selected_task = st.sidebar.radio("Select a task", tasks)
 
-# Main content area based on selected task
-st.header(f"Selected Task: {selected_task}")
+# Main layout based on selected task
+def handle_task_form(task_key, prompt_function, input_label, response_key, example_prompt):
+    # Display form
+    with st.form(key=f"{task_key}_form"):
+        st.markdown(f"<div style='color: #000000; font-weight: bold; background-color: #ffffff; padding: 5px;'>{input_label}</div>", unsafe_allow_html=True)
+        user_input = st.text_area("", key=task_key)
+        submit_button = st.form_submit_button("Get Response")
+        
+        if submit_button:
+            response = prompt_function(user_input)
+            st.session_state['responses'][response_key] = response
+            st.write(response)
+
+    # Display example prompt
+    st.markdown(f"**Example Prompt:** {example_prompt}")
+
+    # Clear button outside of form to avoid modification errors
+    if st.button("Clear"):
+        clear_session_state()
 
 if selected_task == "📚 Personalized Learning Assistant":
-    topic = st.text_input("Enter the topic for the learning plan:")
-    if st.button("Generate Learning Plan"):
-        st.session_state['responses']["personalized_learning_assistant"] = personalized_learning_assistant(topic)
-        st.write(st.session_state['responses']["personalized_learning_assistant"])
+    st.header("Personalized Learning Assistant")
+    handle_task_form(
+        "personalized_learning_assistant", 
+        personalized_learning_assistant,
+        "Enter the topic you want to learn about",
+        "personalized_learning_assistant",
+        "e.g., 'Create a study plan for learning Machine Learning with Python'"
+    )
 
 elif selected_task == "🤖 AI Coding Mentor":
-    code_snippet = st.text_area("Paste your AI code snippet here:")
-    if st.button("Review Code"):
-        st.session_state['responses']["ai_coding_mentor"] = ai_coding_mentor(code_snippet)
-        st.write(st.session_state['responses']["ai_coding_mentor"])
+    st.header("AI Coding Mentor")
+    handle_task_form(
+        "ai_coding_mentor",
+        ai_coding_mentor,
+        "Paste your AI code snippet here",
+        "ai_coding_mentor",
+        "e.g., 'Suggest improvements for this AI code snippet: ...'"
+    )
 
 elif selected_task == "📄 Smart Document Summarizer":
-    document_text = st.text_area("Paste your document text here:")
-    if st.button("Summarize Document"):
-        st.session_state['responses']["smart_document_summarizer"] = smart_document_summarizer(document_text)
-        st.write(st.session_state['responses']["smart_document_summarizer"])
+    st.header("Smart Document Summarizer")
+    handle_task_form(
+        "smart_document_summarizer",
+        smart_document_summarizer,
+        "Paste the document text here",
+        "smart_document_summarizer",
+        "e.g., 'Summarize this document: ...'"
+    )
 
 elif selected_task == "🗓️ Interactive Study Planner":
-    exam_schedule = st.text_area("Enter your exam schedule details:")
-    if st.button("Generate Study Plan"):
-        st.session_state['responses']["interactive_study_planner"] = interactive_study_planner(exam_schedule)
-        st.write(st.session_state['responses']["interactive_study_planner"])
-
-elif selected_task == "❓ Real-Time Q&A Support":
-    question = st.text_input("Ask your academic question:")
-    if st.button("Get Answer"):
-        st.session_state['responses']["real_time_qa_support"] = real_time_qa_support(question)
-        st.write(st.session_state['responses']["real_time_qa_support"])
-
-elif selected_task == "💬 Mental Health Check-In":
-    feelings = st.text_area("Describe your current feelings:")
-    if st.button("Get Advice"):
-        st.session_state['responses']["mental_health_check_in"] = mental_health_check_in(feelings)
-        st.write(st.session_state['responses']["mental_health_check_in"])
-
-# Layout for buttons side-by-side
-col1, col2 = st.columns([3, 1])
-with col1:
-    if st.button("Clear Session"):
-        clear_session_state()
-        st.write("Session cleared.")
+    st.header("Interactive Study Planner")
+    handle_task_form(
+        "interactive_study_planner",
+        interactive_study_planner,
+        "Enter your exam schedule here",
+        "interactive_study_planner",
+        "e.g., 'Create a study plan for these exams: ...'"
+    )
+
+elif selected_task == "💬 Real-Time Q&A Support":
+    st.header("Real-Time Q&A Support")
+    handle_task_form(
+        "real_time_qa_support",
+        real_time_qa_support,
+        "Ask your academic question here",
+        "real_time_qa_support",
+        "e.g., 'Answer this question: ...'"
+    )
+
+elif selected_task == "🧠 Mental Health Check-In":
+    st.header("Mental Health Check-In")
+    handle_task_form(
+        "mental_health_check_in",
+        mental_health_check_in,
+        "Share your feelings or concerns",
+        "mental_health_check_in",
+        "e.g., 'Feeling stressed? Here's some advice: ...'"
+    )
 
-# Footer
+# Display footer
 st.markdown("""
     <div class="footer">
-        <a href="https://www.yourwebsite.com">Home</a>
-        <a href="https://www.yourwebsite.com/about">About</a>
-        <a href="https://www.yourwebsite.com/contact">Contact</a>
-        <a href="https://www.yourwebsite.com/privacy">Privacy</a>
+        <a href="https://twitter.com/yourprofile" target="_blank"><i class="fab fa-twitter"></i></a>
+        <a href="https://github.com/yourprofile" target="_blank"><i class="fab fa-github"></i></a>
+        <a href="mailto:your.email@example.com"><i class="fas fa-envelope"></i></a>
     </div>
 """, unsafe_allow_html=True)