UI Improvement for the search university page.

app.py

@@ -157,51 +157,37 @@ def main():
     # Main header
     st.markdown("""
     <div class="main-header">
-        <h1>🎓 PanSea University Search</h1>
-        <p>AI-Powered Study Search Platform for ASEAN Universities</p>
+        <h1>🎓 Top.Edu</h1>
+        <h5>Unlock ASEAN Education with AI-Powered Search</h5>
     </div>
     """, unsafe_allow_html=True)
     
     # Sidebar
     with st.sidebar:
-        st.header("📋 Navigation")
-        page = st.selectbox(
-            "Choose a page:",
-            ["🔍 Search Universities", "📄 Upload Documents", "🗂 Manage Documents", "ℹ️ About"]
-        )
-        
-        # Show embedding model info
-        st.markdown("---")
-        try:
-            from rag_system import RAGSystem
-            temp_rag = RAGSystem()
-            if hasattr(temp_rag.embeddings, 'model') and temp_rag.embeddings.model:
-                st.markdown("""
-                <div class='language-selection'>
-                    <h5 style='margin: 0; color: #9b59b6;'>🔧 Embedding Model</h5>
-                    <p style='margin: 5px 0; font-size: 0.9em;'>BGE-small-en-v1.5</p>
-                </div>
-                """, unsafe_allow_html=True)
-            else:
-                st.markdown("""
-                <div class='language-selection'>
-                    <h5 style='margin: 0; color: #9b59b6;'>🔧 Embedding Model</h5>
-                    <p style='margin: 5px 0; font-size: 0.9em;'>OpenAI Ada-002</p>
-                </div>
-                """, unsafe_allow_html=True)
-        except:
-            pass
-    
+
+        # Define the pages
+        pages = ["🔍 Search Universities", "📄 Upload Documents", "🗂 Manage Documents", "ℹ️ About Top.Edu"]
+
+        # Initialize session state if needed
+        if "current_page" not in st.session_state:
+            st.session_state.current_page = pages[0]
+
+        # Create buttons for each page
+        for p in pages:
+            if st.button(p, use_container_width=True):
+                st.session_state.current_page = p
+
     # Main content based on selected page
-    if page == "📄 Upload Documents":
+    if st.session_state.current_page == "📄 Upload Documents":
         upload_documents_page()
-    elif page == "🗂 Manage Documents":
+    elif st.session_state.current_page == "🗂 Manage Documents":
         manage_documents_page()
-    elif page == "ℹ️ About":
+    elif st.session_state.current_page == "ℹ️ About":
         about_page()
     else:
         search_page()
 
+
 def upload_documents_page():
     st.header("📄 Upload University Documents")
     st.write("Upload PDF documents containing university admission requirements, fees, and program information.")
@@ -338,197 +324,90 @@ def manage_documents_page():
 
 def search_page():
     st.header("🔍 Search University Information")
-    
-    # Language selection
+
+    # --- Language selection ---
     col1, col2 = st.columns([3, 1])
     with col1:
-        st.write("Ask questions about university admissions, requirements, fees, and programs:")
+        st.write("Ask about admissions, fees, scholarships, and programs.")
     with col2:
         response_language = st.selectbox(
-            "Response Language",
-            ["English", "中文 (Chinese)", "Bahasa Malaysia", "ไทย (Thai)", "Bahasa Indonesia", "Tiếng Việt (Vietnamese)"],
+            "Language",
+            ["English", "中文 (Chinese)", "Bahasa Malaysia", "ไทย (Thai)", 
+             "Bahasa Indonesia", "Tiếng Việt (Vietnamese)"],
             key="response_language"
         )
-    
-    # Show language info
+
     language_map = {
         "English": "English",
-        "中文 (Chinese)": "Chinese", 
+        "中文 (Chinese)": "Chinese",
         "Bahasa Malaysia": "Malay",
         "ไทย (Thai)": "Thai",
         "Bahasa Indonesia": "Indonesian",
         "Tiếng Việt (Vietnamese)": "Vietnamese"
     }
-    selected_lang = language_map.get(response_language, "English")
-    
+    selected_lang = language_map[response_language]
+
     if selected_lang != "English":
-        st.info(f"🌐 AI will respond in **{selected_lang}** based on your selection")
-    
-    # Example queries with model indicators
-    st.markdown("**💡 Example queries:**")
-    
-    # Add model selection explanation
-    st.markdown("""
-    <div class='model-info'>
-        <h4 style='margin: 0; color: #3498db;'>🤖 AI Model Selection</h4>
-        <p style='margin: 5px 0;'><strong>🧠 Reasoning Model (SEA-LION v3.5):</strong> Complex university searches with multiple criteria, comparisons, budget constraints</p>
-        <p style='margin: 5px 0;'><strong>⚡ Instruct Model (SEA-LION v3):</strong> Simple questions, translations, definitions, basic information</p>
-        <p style='margin: 5px 0; font-style: italic;'>The system automatically chooses the best model for your query!</p>
-    </div>
-    """, unsafe_allow_html=True)
-    
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        st.markdown("**🧠 Complex Queries (Uses Reasoning Model):**")
+        st.info(f"🌐 Responses will be in **{selected_lang}**")
+
+    # --- Query input ---
+    query = st.text_area(
+        "Your question:",
+        height=80,
+        placeholder="e.g., Master's in Malaysia under 40,000 RMB/year",
+    )
+
+    # --- Example queries ---
+    with st.expander("💡 See Example Queries"):
+        tab1, tab2 = st.tabs(["🧠 Complex Queries", "⚡ Simple Queries"])
+        
         complex_examples = [
             "Show me universities in Malaysia for master's degrees with tuition under 40,000 RMB per year",
             "专科毕业，无雅思，想在马来西亚读硕士，学费不超过4万人民币/年",
             "Compare engineering programs in Thailand and Singapore under $15,000 per year",
             "Find MBA programs in ASEAN with GMAT requirements and scholarships available"
         ]
-        for example in complex_examples:
-            st.markdown(f"• {example}")
-        
-    with col2:
-        st.markdown("**⚡ Simple Queries (Uses Instruct Model):**")
         simple_examples = [
             "What does IELTS stand for?",
             "Translate 'application deadline' to Chinese",
-            "What is the difference between bachelor and master degree?", 
+            "What is the difference between bachelor and master degree?",
             "How to say 'university' in Thai?"
         ]
-        for example in simple_examples:
-            st.markdown(f"• {example}")
-    
-    st.markdown("---")  # Separator line
-    
-    # Query input - main input field (always available)
-    query = st.text_area(
-        "Your question:", 
-        height=100, 
-        placeholder="e.g., What are the admission requirements for computer science programs in Singapore?",
-        help="Type your question here or select an example below to get started."
-    )
-    
-    # Show search status
-    if query.strip():
-        st.success("✅ Ready to search! Click the search button when you're ready.")
-    else:
-        st.info("💭 Enter your question in the text box above to start searching.")
-    
-    # Optional: Quick example selection (just for convenience)
-    with st.expander("💡 Example Queries (Click to Use)"):
-        # Combine all examples
-        all_examples = complex_examples + simple_examples
-        
-        col1, col2 = st.columns(2)
-        with col1:
-            st.markdown("**🧠 Complex Examples:**")
-            for example in complex_examples:
-                if st.button(example[:60] + "...", key=f"ex_{hash(example)}", help=f"Click to use: {example}"):
-                    st.session_state.example_query = example
-                    
-        with col2:
-            st.markdown("**⚡ Simple Examples:**")  
-            for example in simple_examples:
-                if st.button(example[:60] + "...", key=f"ex_{hash(example)}", help=f"Click to use: {example}"):
-                    st.session_state.example_query = example
-    
-    # Use selected example if any
-    if hasattr(st.session_state, 'example_query') and st.session_state.example_query:
+
+        for ex in complex_examples:
+            if tab1.button(ex, key=f"complex_{hash(ex)}"):
+                st.session_state.example_query = ex
+
+        for ex in simple_examples:
+            if tab2.button(ex, key=f"simple_{hash(ex)}"):
+                st.session_state.example_query = ex
+
+    # --- Use clicked example ---
+    if 'example_query' in st.session_state:
         query = st.session_state.example_query
-        st.info(f"📝 Using example: {query[:100]}...")
-        # Clear the example after use
+        st.info(f"📝 Using example: {query}")
         del st.session_state.example_query
-    
-    # Additional filters
-    with st.expander("🔧 Advanced Filters (Optional)"):
+        # Optionally auto-trigger search
+
+    # --- Optional filters ---
+    with st.expander("🔧 Advanced Filters"):
         col1, col2, col3 = st.columns(3)
-        with col1:
-            budget_range = st.select_slider(
-                "Budget Range (USD/year)",
-                options=["Any", "<10k", "10k-20k", "20k-30k", "30k-40k", ">40k"],
-                value="Any"
-            )
-        with col2:
-            study_level = st.multiselect(
-                "Study Level",
-                ["Diploma", "Bachelor", "Master", "PhD"],
-                default=[]
-            )
-        with col3:
-            preferred_countries = st.multiselect(
-                "Preferred Countries",
-                ["Singapore", "Malaysia", "Thailand", "Indonesia", "Philippines", "Vietnam", "Brunei"],
-                default=[]
-            )
-    
-    # Search button - enabled as soon as there's text in the query
-    search_disabled = not query.strip()
-    button_text = "🔍 Search" if not search_disabled else "🔍 Search (Enter a question first)"
-    
-    if st.button(button_text, type="primary", disabled=search_disabled):
-        if not query.strip():
-            st.error("Please enter a question.")
-            return
-        
-        # Get the language code for processing
-        language_map = {
-            "English": "English",
-            "中文 (Chinese)": "Chinese", 
-            "Bahasa Malaysia": "Malay",
-            "ไทย (Thai)": "Thai",
-            "Bahasa Indonesia": "Indonesian",
-            "Tiếng Việt (Vietnamese)": "Vietnamese"
-        }
-        language_code = language_map.get(response_language, "English")
+        budget_range = col1.select_slider(
+            "Budget (USD/year)", 
+            options=["Any", "<10k", "10k-20k", "20k-30k", "30k-40k", ">40k"],
+            value="Any"
+        )
+        study_level = col2.multiselect(
+            "Study Level", ["Diploma", "Bachelor", "Master", "PhD"]
+        )
+        preferred_countries = col3.multiselect(
+            "Preferred Countries",
+            ["Singapore", "Malaysia", "Thailand", "Indonesia", "Philippines", "Vietnam", "Brunei"]
+        )
 
-        with st.spinner("Searching for information..."):
-            try:
-                # Initialize RAG system
-                rag_system = RAGSystem()
-                
-                # Show which model will be used
-                from rag_system import classify_query_type
-                query_type = classify_query_type(query)
-                
-                if query_type == "complex":
-                    st.info("🧠 **Using SEA-LION Reasoning Model (v3.5)** - Complex query detected")
-                else:
-                    st.info("⚡ **Using SEA-LION Instruct Model (v3)** - Simple query/translation detected")
-                
-                # Show translation status if not English
-                if response_language != "English":
-                    st.info(f"🌐 **Translating response to {response_language}**")
-                
-                # Add filters to query if specified
-                enhanced_query = query
-                if budget_range != "Any" or study_level or preferred_countries:
-                    filters = []
-                    if budget_range != "Any":
-                        filters.append(f"budget range: {budget_range}")
-                    if study_level:
-                        filters.append(f"study levels: {', '.join(study_level)}")
-                    if preferred_countries:
-                        filters.append(f"countries: {', '.join(preferred_countries)}")
-                    
-                    enhanced_query += f"\n\nAdditional filters: {'; '.join(filters)}"
-                
-                # Get response
-                result = rag_system.query(enhanced_query, language_code)
-                
-                if result:
-                    # Save query result for sharing
-                    save_query_result(result)
-                    
-                    # Display results
-                    display_query_result(result, show_share_link=True)
-                else:
-                    st.error("No results found. Try rephrasing your question or upload more documents.")
-            
-            except Exception as e:
-                st.error(f"Error searching: {str(e)}")
+    # --- Search button ---
+    if st.button("🔍 Search", type="primary", disabled=not query.strip()):
+        st.success("Searching...")  # Placeholder for your RAGSystem logic
 
 def display_query_result(result, show_share_link=False):
     """Display query results in a formatted way."""