Update app.py

app.py

@@ -30,9 +30,15 @@ suggested_questions = [
     "What was the revenue growth of"
 ]
 
-# Streamlit UI
-st.title("Question Completion Model")
-st.write("Enter a partial question related to financial statements, corporate governance, or company reports, and the app will intelligently complete it based on learned patterns!")
+# Sidebar Settings
+st.sidebar.header("🔧 Settings")
+max_length = st.sidebar.slider("Max Length", 30, 100, 60, step=5)
+num_return_sequences = st.sidebar.slider("Number of Completions", 1, 5, 3)
+sampling = st.sidebar.checkbox("Enable Sampling", value=True)
+
+# Main UI
+st.title("🤖 Question Completion Model")
+st.write("Enter a partial question related to **financial statements, corporate governance, or company reports**, and the model will intelligently complete it!")
 
 # Session state for input question
 if "partial_question" not in st.session_state:
@@ -43,49 +49,37 @@ def update_input(selected_question):
     st.session_state.partial_question = selected_question
     st.rerun()
 
-# Display suggested partial questions
-st.write("### Suggested Partial Questions:")
-cols = st.columns(5)
+# Display suggested partial questions in sidebar
+st.sidebar.subheader("💡 Suggested Partial Questions")
 for i, question in enumerate(suggested_questions):
-    if cols[i % 5].button(question, key=f"suggested_{i}"):  # Place buttons in columns
+    if st.sidebar.button(question, key=f"suggested_{i}"):  
         update_input(question)
 
 # Text input box
+st.subheader("✏️ Enter Partial Question")
 partial_question = st.text_input("Enter a partial question:", st.session_state.partial_question)
 
-# Button to generate 3 completed questions
-if st.button("Complete Question"):
+# Generate button
+if st.button("🚀 Complete Question"):
     if partial_question.strip():
-        outputs = question_completion_pipeline(
-            partial_question,
-            max_length=60,
-            num_return_sequences=3,  # Generate 3 different completions
-            do_sample=True,
-            truncation=True  
-        )
-
-        st.write("### Completed Questions:")
-        for output in outputs:
-            completed_question = output["generated_text"]
-            st.markdown(f"**{completed_question}**")  # Display non-clickable text
-
+        with st.spinner("⏳ Generating completed questions... Please wait!"):
+            outputs = question_completion_pipeline(
+                partial_question,
+                max_length=max_length,
+                num_return_sequences=num_return_sequences,
+                do_sample=sampling,
+                truncation=True  
+            )
+
+        st.subheader("✅ Completed Questions")
+        completed_texts = [output["generated_text"] for output in outputs]
+
+        # Display completed questions inside an expander
+        with st.expander("🔍 View Generated Questions"):
+            for idx, completed_question in enumerate(completed_texts):
+                st.markdown(f"**{idx+1}. {completed_question}**")
+        
+        # Copy option
+        st.text_area("📋 Copy Completed Questions:", "\n".join(completed_texts), height=150)
     else:
-        st.warning("Please enter a partial question.")
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+        st.warning("⚠️ Please enter a partial question.")