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, text area, and flashcards
+# Custom CSS for the app
 st.markdown("""
     <style>
         .main .block-container {
@@ -15,56 +15,19 @@ 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;
         }
-        .flashcard-container {
-            display: flex;
-            flex-wrap: wrap;
-            gap: 1rem;
-        }
-        .flashcard {
-            width: 250px;
-            height: 200px;
-            border: 1px solid blue;
-            border-radius: 5px;
-            font-size: 16px;
-            color: blue;
-            background-color: #f9f9f9;
-            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);
+        .question-list {
+            margin-top: 1rem;
         }
-        .flashcard-front, .flashcard-back {
-            position: absolute;
-            width: 100%;
-            height: 100%;
-            backface-visibility: hidden;
-            display: flex;
-            align-items: center;
-            justify-content: center;
+        .question {
             padding: 1rem;
-        }
-        .flashcard-front {
-            background-color: #fff;
-        }
-        .flashcard-back {
+            border: 1px solid #ddd;
+            border-radius: 5px;
             background-color: #f9f9f9;
-            transform: rotateY(180deg);
+            margin-bottom: 0.5rem;
         }
     </style>
 """, unsafe_allow_html=True)
@@ -72,18 +35,14 @@ st.markdown("""
 # 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 = ""
+if 'questions' not in st.session_state:
+    st.session_state.questions = ""
 
 # Display the title and introduction
-st.title("AI71 Text Processing Tool")
+st.title("InsightStarfleet")
 st.write("""
-    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.
+    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.
 """)
 
 # Input area for text
@@ -93,7 +52,7 @@ input_text = st.text_area(
     value=st.session_state.input_text,
     height=200,
     key="input_text",
-    help="Paste or type your text here for processing."
+    help="Paste or type your text here to generate questions."
 )
 
 # Update session state if input text changes
@@ -104,81 +63,30 @@ if input_text != st.session_state.input_text:
 if st.session_state.input_text:
     AI71_API_KEY = get_ai71_api_key()
 
-    st.subheader("Text Processing")
+    st.subheader("Generate Questions")
     
     # Caching API call function
     @st.cache_data
     def generate_response(system_message, user_message):
         return _generate_response(system_message, user_message)
 
-    # 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:"
-        st.session_state.flashcards = generate_response(flashcards_prompt, st.session_state.input_text)
-
-# Results Tabs
-tabs = st.tabs(["Summary", "Main Points", "Flashcards"])
-
-with tabs[0]:
-    if st.session_state.summary:
-        st.subheader("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"
-        )
-
-with tabs[1]:
-    if st.session_state.main_points:
-        st.subheader("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"
-        )
+    # 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)
 
-with tabs[2]:
-    if st.session_state.flashcards:
-        st.subheader("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:
-            question, answer = card.split('\n', 1)  # Assuming each card is in "question\nanswer" format
-            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)
-        st.markdown('</div>', unsafe_allow_html=True)
-        st.download_button(
-            label="Download Flashcards",
-            data=st.session_state.flashcards,
-            file_name="flashcards.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)
+    st.markdown('</div>', unsafe_allow_html=True)
+    st.download_button(
+        label="Download Questions",
+        data=st.session_state.questions,
+        file_name="questions.txt",
+        mime="text/plain"
+    )