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 for the app
+# Custom CSS to style the page, text area, and flashcards
 st.markdown("""
     <style>
         .main .block-container {
@@ -15,19 +15,56 @@ st.markdown("""
             text-align: center;
             margin-bottom: 1rem;
         }
+        .section-title {
+            margin-top: 2rem;
+            font-size: 1.5rem;
+        }
         .text-area {
             margin-top: 1rem;
             margin-bottom: 1rem;
         }
-        .question-list {
-            margin-top: 1rem;
+        .flashcard-container {
+            display: flex;
+            flex-wrap: wrap;
+            gap: 1rem;
         }
-        .question {
-            padding: 1rem;
-            border: 1px solid #ddd;
+        .flashcard {
+            width: 250px;
+            height: 200px;
+            border: 1px solid blue;
             border-radius: 5px;
+            font-size: 16px;
+            color: black;
             background-color: #f9f9f9;
-            margin-bottom: 0.5rem;
+            perspective: 1000px;
+            cursor: pointer;
+        }
+        .flashcard-inner {
+            position: relative;
+            width: 100%;
+            height: 100%;
+            transition: transform 0.6s;
+            transform-style: preserve-3d;
+        }
+        .flashcard:hover .flashcard-inner {
+            transform: rotateY(180deg);
+        }
+        .flashcard-front, .flashcard-back {
+            position: absolute;
+            width: 100%;
+            height: 100%;
+            backface-visibility: hidden;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            padding: 1rem;
+        }
+        .flashcard-front {
+            background-color: #fff;
+        }
+        .flashcard-back {
+            background-color: #f9f9f9;
+            transform: rotateY(180deg);
         }
     </style>
 """, unsafe_allow_html=True)
@@ -35,14 +72,18 @@ st.markdown("""
 # Initialize session state variables if not already set
 if 'input_text' not in st.session_state:
     st.session_state.input_text = ""
-if 'questions' not in st.session_state:
-    st.session_state.questions = ""
+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("InsightStarfleet")
 st.write("""
-    Welcome to the InsightStarfleet! This application helps you summarize, and generate main points and flash cards based on the input text.
-    Enter your text below, and click the button to generate questions.
+    Welcome to the InsightStarfleet! 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.
 """)
 
 # Input area for text
@@ -52,7 +93,7 @@ input_text = st.text_area(
     value=st.session_state.input_text,
     height=200,
     key="input_text",
-    help="Paste or type your text here to generate questions."
+    help="Paste or type your text here for processing."
 )
 
 # Update session state if input text changes
@@ -63,30 +104,88 @@ if input_text != st.session_state.input_text:
 if st.session_state.input_text:
     AI71_API_KEY = get_ai71_api_key()
 
-    st.subheader("Generate Questions")
+    st.subheader("Text Processing")
     
     # Caching API call function
     @st.cache_data
     def generate_response(system_message, user_message):
         return _generate_response(system_message, user_message)
 
-    # Button for generating questions
-    if st.button("Generate Questions"):
-        questions_prompt = "Generate questions based on the following text:"
-        st.session_state.questions = generate_response(questions_prompt, st.session_state.input_text)
+    # Buttons for processing
+    if st.button("Generate Summary"):
+        summary_prompt = "Summarize the following text into a brief summary:"
+        st.session_state.summary = generate_response(summary_prompt, st.session_state.input_text)
+
+    if st.button("Extract Main Points"):
+        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)
+
+    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)
+
+# Combined Results
+st.subheader("Results")
+
+# Summary
+if st.session_state.summary:
+    st.markdown("### Summary")
+    st.text_area(
+        "Summary",
+        value=st.session_state.summary,
+        height=200,
+        key="summary_results",
+        help="Generated summary."
+    )
+    st.download_button(
+        label="Download Summary",
+        data=st.session_state.summary,
+        file_name="summary.txt",
+        mime="text/plain"
+    )
+
+# 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")])
+    st.markdown(f"### Main Points\n{bullet_points}", unsafe_allow_html=True)
+    st.download_button(
+        label="Download Main Points",
+        data=bullet_points,
+        file_name="main_points.txt",
+        mime="text/plain"
+    )
 
-# Display generated questions
-if st.session_state.questions:
-    st.subheader("Generated Questions")
-    question_list = st.session_state.questions.split('\n')  # Assuming questions are separated by newlines
-    st.markdown('<div class="question-list">', unsafe_allow_html=True)
-    for question in question_list:
-        if question.strip():
-            st.markdown(f"<div class='question'>{question.strip()}</div>", unsafe_allow_html=True)
+# Flashcards
+if st.session_state.flashcards:
+    st.markdown("### Flashcards")
+    flashcards = st.session_state.flashcards.split("\n\n")  # Assuming flashcards are separated by double newlines
+    st.markdown('<div class="flashcard-container">', unsafe_allow_html=True)
+    for card in flashcards:
+        try:
+            # Check if card contains both question and answer
+            if '\n' in card:
+                question, answer = card.split('\n', 1)
+                st.markdown(f"""
+                    <div class="flashcard">
+                        <div class="flashcard-inner">
+                            <div class="flashcard-front">
+                                <p>{question.strip()}</p>
+                            </div>
+                            <div class="flashcard-back">
+                                <p>{answer.strip()}</p>
+                            </div>
+                        </div>
+                    </div>
+                """, unsafe_allow_html=True)
+            else:
+                st.error("Invalid flashcard format. Each flashcard should be in 'question\\nanswer' format.")
+        except Exception as e:
+            st.error(f"Error processing flashcard: {e}")
     st.markdown('</div>', unsafe_allow_html=True)
     st.download_button(
-        label="Download Questions",
-        data=st.session_state.questions,
-        file_name="questions.txt",
+        label="Download Flashcards",
+        data=st.session_state.flashcards,
+        file_name="flashcards.txt",
         mime="text/plain"
     )