Update app.py

app.py

@@ -5,7 +5,7 @@ from ai_utils import generate_response as _generate_response
 # Set the page configuration to wide layout
 st.set_page_config(layout="wide")
 
-# Custom CSS to style the page
+# Custom CSS to style the page and text area
 st.markdown("""
     <style>
         .main .block-container {
@@ -30,14 +30,23 @@ st.markdown("""
             margin: 0.5rem 0;
             background-color: #f9f9f9;
         }
+        .stTextArea textarea {
+            font-size: 16px;
+            border-radius: 5px;
+            border: 2px solid #ddd;
+        }
     </style>
 """, unsafe_allow_html=True)
 
-# Initialize session state variables
-session_vars = ['input_text', 'summary', 'main_points', 'flashcards', 'results']
-for var in session_vars:
-    if var not in st.session_state:
-        st.session_state[var] = ""
+# Initialize session state variables if not already set
+if 'input_text' not in st.session_state:
+    st.session_state.input_text = ""
+if 'summary' not in st.session_state:
+    st.session_state.summary = ""
+if 'main_points' not in st.session_state:
+    st.session_state.main_points = ""
+if 'flashcards' not in st.session_state:
+    st.session_state.flashcards = ""
 
 # Display the title and introduction
 st.title("AI71 Text Processing Tool")
@@ -48,7 +57,17 @@ st.write("""
 
 # Input area for text
 st.subheader("Enter Your Text")
-st.session_state.input_text = st.text_area("Enter text below:", height=200, key="input_text")
+input_text = st.text_area(
+    label="Enter text below:",
+    value=st.session_state.input_text,
+    height=200,
+    key="input_text",
+    help="Paste or type your text here for processing."
+)
+
+# Update session state if input text changes
+if input_text != st.session_state.input_text:
+    st.session_state.input_text = input_text
 
 # Text Processing
 if st.session_state.input_text:
@@ -68,9 +87,12 @@ if st.session_state.input_text:
         st.text_area("Summary", value=st.session_state.summary, height=200, key="summary", help="Summary of the input text.")
 
     if st.button("Extract Main Points"):
-        main_points_prompt = "Extract the main points from the following text:"
+        main_points_prompt = "Extract the main points from the following text and format them as a bulleted list:"
         st.session_state.main_points = generate_response(main_points_prompt, st.session_state.input_text)
-        st.text_area("Main Points", value=st.session_state.main_points, height=200, key="main_points", help="Main points extracted from the input text.")
+        # Format main points as a bulleted list
+        if st.session_state.main_points:
+            bullet_points = "\n".join([f"- {point.strip()}" for point in st.session_state.main_points.split("\n")])
+            st.markdown(f"### Main Points\n{bullet_points}", unsafe_allow_html=True)
 
     if st.button("Generate Flashcards"):
         flashcards_prompt = "Create flashcards with questions and answers based on the following text:"
@@ -93,14 +115,19 @@ if st.session_state.summary or st.session_state.main_points or st.session_state.
     st.text_area("Summary", value=st.session_state.summary, height=200, key="summary_results", help="Generated summary.")
 
     st.subheader("Main Points")
-    st.text_area("Main Points", value=st.session_state.main_points, height=200, key="main_points_results", help="Extracted main points.")
+    if st.session_state.main_points:
+        bullet_points = "\n".join([f"- {point.strip()}" for point in st.session_state.main_points.split("\n")])
+        st.markdown(f"### Main Points\n{bullet_points}", unsafe_allow_html=True)  # Displaying the main points as a bulleted list
 
     st.subheader("Flashcards")
-    st.markdown("""
-        <div class="flashcard">
-            {flashcards}
-        </div>
-    """, unsafe_allow_html=True)
+    if st.session_state.flashcards:
+        flashcards = st.session_state.flashcards.split("\n\n")  # Assuming flashcards are separated by double newlines
+        for card in flashcards:
+            st.markdown(f"""
+                <div class="flashcard">
+                    {card}
+                </div>
+            """, unsafe_allow_html=True)
     
     st.subheader("Download Results")
     results = f"Summary:\n{st.session_state.summary}\n\nMain Points:\n{st.session_state.main_points}\n\nFlashcards:\n{st.session_state.flashcards}"