Update app.py

app.py

@@ -25,29 +25,64 @@ def call_groq_api(prompt):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Define functions for each tool
+# Define functions for each tool with few-shot examples
 def personalized_learning_assistant(topic):
-    prompt = f"Provide a personalized learning plan for the topic: {topic}"
+    prompt = f"""
+    Provide a personalized learning plan for the topic: {topic}.
+    Example 1: For 'Machine Learning': 'Create a plan with courses on supervised learning, unsupervised learning, and neural networks. Include practical exercises and projects.'
+    Example 2: For 'Data Science': 'Suggest a plan with a focus on data analysis, visualization, and statistical modeling. Recommend tools like Python, R, and SQL.'
+    Example 3: For 'Web Development': 'Outline a plan that covers front-end and back-end development, including HTML, CSS, JavaScript, and server-side technologies.'
+    Example 4: For '{topic}':"""
     return call_groq_api(prompt)
 
 def ai_coding_mentor(code_snippet):
-    prompt = f"Review this AI code snippet and provide suggestions or improvements:\n\n{code_snippet}"
+    prompt = f"""
+    Review this AI code snippet and provide suggestions or improvements:
+    {code_snippet}
+    Example 1: 'In the provided code, consider using a different activation function to improve performance.'
+    Example 2: 'The code can be optimized by reducing redundant calculations and enhancing readability.'
+    Example 3: 'Add comments to explain complex sections of the code for better understanding.'
+    Example 4: 'For the provided snippet, suggest improvements or fixes:'''
     return call_groq_api(prompt)
 
 def smart_document_summarizer(document_text):
-    prompt = f"Summarize the following document:\n\n{document_text}"
+    prompt = f"""
+    Summarize the following document:
+    {document_text}
+    Example 1: 'The document discusses the impact of climate change on global agriculture, emphasizing the need for sustainable practices.'
+    Example 2: 'It provides an overview of recent advancements in AI technology and its applications in various fields.'
+    Example 3: 'The text outlines the historical development of renewable energy sources and their future potential.'
+    Example 4: 'Summarize this document:'''
     return call_groq_api(prompt)
 
 def interactive_study_planner(exam_schedule):
-    prompt = f"Create an interactive study plan based on this exam schedule:\n\n{exam_schedule}"
+    prompt = f"""
+    Create an interactive study plan based on this exam schedule:
+    {exam_schedule}
+    Example 1: 'For an exam schedule with subjects A, B, and C, create a plan that allocates study time for each subject and includes breaks.'
+    Example 2: 'Plan should include daily study goals and revision sessions leading up to the exams.'
+    Example 3: 'Suggest a study plan that balances subject preparation with relaxation to avoid burnout.'
+    Example 4: 'Based on the provided schedule, create a study plan:'''
     return call_groq_api(prompt)
 
 def real_time_qa_support(question):
-    prompt = f"Provide an answer to the following academic question:\n\n{question}"
+    prompt = f"""
+    Provide an answer to the following academic question:
+    {question}
+    Example 1: 'Question: What is the capital of France? Answer: Paris.'
+    Example 2: 'Question: Explain the theory of relativity. Answer: The theory of relativity, developed by Albert Einstein, includes two theories: special relativity and general relativity, explaining the relationship between space, time, and gravity.'
+    Example 3: 'Question: What is the process of photosynthesis? Answer: Photosynthesis is the process by which green plants use sunlight to synthesize foods with the help of chlorophyll, water, and carbon dioxide.'
+    Example 4: 'Answer this question:'''
     return call_groq_api(prompt)
 
 def mental_health_check_in(feelings):
-    prompt = f"Provide some advice based on these feelings:\n\n{feelings}"
+    prompt = f"""
+    Provide some advice based on these feelings:
+    {feelings}
+    Example 1: 'Feeling stressed? Try practicing mindfulness and deep breathing exercises to calm your mind.'
+    Example 2: 'If you're feeling anxious, consider talking to a trusted friend or counselor for support.'
+    Example 3: 'Feeling overwhelmed? Break your tasks into smaller steps and focus on completing them one at a time.'
+    Example 4: 'Based on these feelings, provide advice:'''
     return call_groq_api(prompt)
 
 # Initialize session state if not already set
@@ -63,11 +98,8 @@ if 'responses' not in st.session_state:
 
 # Function to clear session state values
 def clear_session_state():
-    for key in st.session_state.keys():
-        if key.startswith('responses'):
-            st.session_state[key] = ""
-        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] = ""
+    for key in st.session_state['responses'].keys():
+        st.session_state['responses'][key] = ""
 
 # Add custom styling using Streamlit
 st.markdown("""
@@ -153,97 +185,63 @@ 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 layout based on selected task
-def handle_task_form(task_key, prompt_function, input_label, response_key):
-    # 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)
-
-    # Clear button outside of form to avoid modification errors
-    if st.button("Clear"):
-        clear_session_state()
+# Main content area based on selected task
+st.header(f"Selected Task: {selected_task}")
 
 if selected_task == "📚 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"
-    )
+    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"])
 
 elif selected_task == "🤖 AI Coding Mentor":
-    st.header("AI Coding Mentor")
-    handle_task_form(
-        "ai_coding_mentor", 
-        ai_coding_mentor,
-        "Paste your code snippet here",
-        "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"])
 
 elif selected_task == "📄 Smart Document Summarizer":
-    st.header("Smart Document Summarizer")
-    handle_task_form(
-        "smart_document_summarizer", 
-        smart_document_summarizer,
-        "Paste your document text here",
-        "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"])
 
 elif selected_task == "🗓️ Interactive Study Planner":
-    st.header("Interactive Study Planner")
-    handle_task_form(
-        "interactive_study_planner", 
-        interactive_study_planner,
-        "Enter your exam schedule here",
-        "interactive_study_planner"
-    )
-
-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"
-    )
-
-elif selected_task == "🧠 Mental Health Check-In":
-    st.header("Mental Health Check-In")
-    with st.form(key="mental_health_check_in_form"):
-        st.markdown("<div style='color: #000000; font-weight: bold; background-color: #ffffff; padding: 5px;'>Share how you are feeling today</div>", unsafe_allow_html=True)
-        feelings = st.text_area("", key="mental_health_check_in")
-        submit_button = st.form_submit_button("Get Advice")
-        
-        if submit_button:
-            response = mental_health_check_in(feelings)
-            st.session_state['responses']['mental_health_check_in'] = response
-            st.write(response)
-
-    # Clear button outside of form to avoid modification errors
-    if st.button("Clear"):
+    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.")
 
 # Footer
 st.markdown("""
     <div class="footer">
-        <a href="https://github.com/muhammadibrahim313">GitHub</a> |
-        <a href="https://www.kaggle.com/itshappy">Kaggle</a> |
-        <a href="https://www.linkedin.com/in/muhammad-ibrahim313">LinkedIn</a>
-        <br>
-        <i>EduNexus 2024</i>
+        <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>
     </div>
 """, unsafe_allow_html=True)