Update app.py

app.py

@@ -157,6 +157,7 @@ if selected_menu == "Home":
         "ViDraft-TTS.png",
         "ViDraft-image-remove background.png"
     ]
+    
     image_texts = [
         "Video Template",
         "Video Search",
@@ -164,11 +165,15 @@ if selected_menu == "Home":
         "TTS (Voice)",
         "ViDraft-image-remove background"
     ]
-    for col, img_file, img_text in zip(cols, image_files, image_texts):
-        with col:
-            st.image(img_file, use_column_width=True)
-            st.caption(img_text)  # 파일명을 통해 이미지 아래 텍스트 반영
-            
+
+ # 이미지들을 한 줄에 4장씩 배치
+    for i in range(0, len(image_files), 4):
+        cols = st.columns(4)  # 4개의 열 생성
+        for col, img_file, img_text in zip(cols, image_files[i:i+4], image_texts[i:i+4]):
+            with col:
+                st.image(img_file, use_column_width=True)
+                st.caption(img_text)
+
 
 elif selected_menu == "Free Stock":
     # 'Free Stock' 메뉴 선택 시 처리 로직