Update app.py

app.py

@@ -25,9 +25,9 @@ st.markdown("""
         }
         .flashcard-container {
             display: flex;
-            flex-wrap: wrap;
+            overflow-x: auto; /* Allow horizontal scrolling */
+            white-space: nowrap; /* Prevent wrapping */
             gap: 1rem;
-            overflow-x: auto; /* Allow horizontal scrolling if needed */
         }
         .flashcard {
             width: 250px;
@@ -42,6 +42,7 @@ st.markdown("""
             display: flex;
             align-items: center;
             justify-content: center;
+            flex-shrink: 0; /* Prevent shrinking */
         }
         .flashcard-inner {
             position: relative;
@@ -84,9 +85,9 @@ if 'flashcards' not in st.session_state:
     st.session_state.flashcards = ""
 
 # Display the title and introduction
-st.title("InsightStarfleet")
+st.title("AI71 Text Processing Tool")
 st.write("""
-    Welcome to the InsightStarfleet! This application helps you process text by summarizing content,
+    Welcome to the AI71 Text Processing Tool! This application helps you process text by summarizing content,
     extracting main points, and creating flashcards. Follow the instructions to enter your text and generate content.
 """)
 
@@ -110,7 +111,9 @@ if st.session_state.input_text:
     # Caching API call function
     @st.cache_data
     def generate_response(system_message, user_message):
-        return _generate_response(system_message, user_message)
+        response = _generate_response(system_message, user_message)
+        # Strip any unwanted labels or text
+        return response.strip()
 
     # Buttons for processing
     if st.button("Generate Summary"):
@@ -124,6 +127,10 @@ if st.session_state.input_text:
     if st.button("Generate Flashcards"):
         flashcards_prompt = "Create flashcards with questions and answers based on the following text. Ensure each flashcard includes a question and its corresponding answer, formatted as 'Question: [question]\\nAnswer: [answer]':"
         st.session_state.flashcards = generate_response(flashcards_prompt, st.session_state.input_text)
+else:
+    # Only show generate buttons if text is present
+    if st.button("Generate Summary") or st.button("Extract Main Points") or st.button("Generate Flashcards"):
+        st.error("Please enter text before generating content.")
 
 # Combined Results
 st.subheader("Results")
@@ -148,7 +155,8 @@ if st.session_state.summary:
 # Main Points
 if st.session_state.main_points:
     st.markdown("### Main Points")
-    bullet_points = "\n".join([f"- {point.strip()}" for point in st.session_state.main_points.split("\n")])
+    # Remove any leading or trailing blank lines from main points
+    bullet_points = "\n".join([f"- {point.strip()}" for point in st.session_state.main_points.split("\n") if point.strip()])
     st.markdown(f"### Main Points\n{bullet_points}", unsafe_allow_html=True)
     st.download_button(
         label="Download Main Points",