Update app.py

app.py

@@ -250,16 +250,26 @@ else:
         default_values = example_data["semester_data"][idx*3:(idx+1)*3] if example_data else None  # Fix this line to slice 3 values per semester (Số tín chỉ đạt, Số tín chỉ nợ, GPA)
         
         # Use expander to group each semester's fields in a collapsible box
-        with st.expander(f"📘 {sem}", expanded=True):
-            col1, col2 = st.columns(2)
-            with col1:
-                sotc_qua = st.number_input(f"Số tín chỉ đạt - {sem}", 0, value=default_values[0] if default_values else 0, key=f"tcqua_imp_{sem}")
-                sotcno = st.number_input(f"Số tín chỉ nợ - {sem}", 0, value=default_values[1] if default_values else 0, key=f"tcno_imp_{sem}")
-            with col2:
-                TBCHK = st.number_input(f"Điểm trung bình - {sem}", 0.0, 10.0, value=default_values[2] if default_values else 0.0, step=0.01, key=f"gpa_imp_{sem}")
-            
-            # Collect semester data
-            data += [sotc_qua, sotcno, TBCHK]  
+        if idx < 3:  # First row (3 boxes)
+            with st.expander(f"📘 {sem}", expanded=True):
+                col1, col2, col3 = st.columns(3)  # 3 columns for the first row
+                with col1:
+                    sotc_qua = st.number_input(f"Số tín chỉ đạt - {sem}", 0, value=default_values[0] if default_values else 0, key=f"tcqua_imp_{sem}")
+                with col2:
+                    sotcno = st.number_input(f"Số tín chỉ nợ - {sem}", 0, value=default_values[1] if default_values else 0, key=f"tcno_imp_{sem}")
+                with col3:
+                    TBCHK = st.number_input(f"Điểm trung bình - {sem}", 0.0, 10.0, value=default_values[2] if default_values else 0.0, step=0.01, key=f"gpa_imp_{sem}")
+        else:  # Second row (2 boxes)
+            with st.expander(f"📘 {sem}", expanded=True):
+                col1, col2 = st.columns(2)  # 2 columns for the second row
+                with col1:
+                    sotc_qua = st.number_input(f"Số tín chỉ đạt - {sem}", 0, value=default_values[0] if default_values else 0, key=f"tcqua_imp_{sem}")
+                    sotcno = st.number_input(f"Số tín chỉ nợ - {sem}", 0, value=default_values[1] if default_values else 0, key=f"tcno_imp_{sem}")
+                with col2:
+                    TBCHK = st.number_input(f"Điểm trung bình - {sem}", 0.0, 10.0, value=default_values[2] if default_values else 0.0, step=0.01, key=f"gpa_imp_{sem}")
+        
+        # Collect semester data
+        data += [sotc_qua, sotcno, TBCHK]  
     
     final_features = np.array(data).reshape(1, -1)