Update app.py

app.py

@@ -1,13 +1,23 @@
 import os
-os.environ["GROQ_API_KEY"] = "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"
-
-
-import streamlit as st
-import os
 from groq import Groq
+import streamlit as st
+
+# Set up Groq API client
+client = Groq(api_key=os.environ.get("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"))
 
-# Set up Groq API
-client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+# Predefined topics
+topics = [
+    "The impact of technology on education.",
+    "The importance of mental health awareness.",
+    "How social media affects communication skills.",
+    "The benefits of reading for personal growth.",
+    "Climate change and its global impact.",
+    "The role of artificial intelligence in everyday life.",
+    "The significance of physical fitness in modern life.",
+    "Why we should promote gender equality in the workplace.",
+    "The effects of globalization on local cultures.",
+    "The importance of renewable energy sources."
+]
 
 # App Title
 st.title("📘 Writing Assistant for IELTS/DET/TOEFL")
@@ -40,21 +50,21 @@ if "plan" in st.session_state:
     st.markdown(f"### Selected Plan: **{st.session_state.plan}**")
     if "current_day" not in st.session_state:
         st.session_state.current_day = 1
-    st.markdown(f"**Day {st.session_state.current_day}** Topic: Write an essay on *'The impact of technology on education.'*")
+    st.markdown(f"**Day {st.session_state.current_day}** Topic: Write an essay on *'{topics[st.session_state.current_day - 1]}'*")
 
 # Essay Submission Section
 st.markdown("### Submit Your Essay")
-user_input = st.text_area(
-    "Write your essay here (minimum 150 words):", height=300
-)
+user_input = st.text_area("Write your essay here (minimum 150 words):", height=300)
+
 if st.button("Submit for Feedback"):
     if len(user_input.split()) < 150:
         st.error("Please write at least 150 words before submitting.")
     else:
         with st.spinner("Processing your essay..."):
-            # Send the essay to Groq API
+            # Send the essay to Groq API for feedback
             prompt = f"""
-            You are an expert academic writer with 40 years of experience in providing concise but effective feedback. Instead of asking the student to do this and that, you just say replace this with this to improve concisely. 
+            You are an expert academic writer with 40 years of experience in providing concise but effective feedback. 
+            Instead of asking the student to do this and that, you just say replace this with this to improve concisely.
             Analyze the following essay: {user_input}.
             {st.session_state.plan} level user should get feedback specific to their level.
             """
@@ -67,6 +77,7 @@ if st.button("Submit for Feedback"):
             st.markdown("### Feedback")
             st.write(feedback)
 
+            # Provide improved version of the essay
             st.markdown("### Improved Version")
             st.write("Here's how your essay could look with the suggested improvements:")
             improved_prompt = f"Revise the following essay for {st.session_state.plan} level user: {user_input}"
@@ -80,4 +91,6 @@ if st.button("Submit for Feedback"):
 # Navigation for Next Day
 if "plan" in st.session_state and st.button("Next Day"):
     st.session_state.current_day += 1
-    st.experimental_rerun()
+    if st.session_state.current_day > len(topics):  # Reset to day 1 if all days are completed
+        st.session_state.current_day = 1
+    st.rerun()  # Use st.experimental_rerun() to re-run the app and update the day and topic