Update app.py

app.py

@@ -153,7 +153,7 @@ if selected_menu == "Video" and st.session_state['current_sub_menu'] == "Remove
 if selected_menu == "Home":
     st.image("banner2.jpg", use_column_width=True)    
     st.header("Welcome to ViDraft")
-    st.subheader("'Create Contents, ViDraft Value-UP'")
+    st.header("'Create Contents, ViDraft Value-UP'")
 
     # 환영 메시지 및 서비스 설명
     welcome_text = """
@@ -169,18 +169,41 @@ if selected_menu == "Home":
     Join us to transform your creative ideas into reality with cutting-edge technology designed for creators, by creators.
 
     -Contact: arxivgpt@gmail.com  / Powered by ArXivGPT, Huggingface, Gradio, Streamlit
-    """
-    
+    """    
     st.write(welcome_text)
 
     # 이미지들을 한 줄에 3장씩 배치
+    image_files = [
+        "ViDraft-Video-Templet.png",
+        "ViDraft-Video-search.png",
+        "ViDraft-Image-search.png",
+        "ViDraft-TTS.png",
+        "ViDraft-image-remove background.png",
+        "ViDraft-video-remove-back.png",
+        "ViDraft-video-comp.png",
+        "ViDraft-image-comp.png"
+    ]
+    
+    image_texts = [
+        "Video Template(OpenAI SORA)",
+        "Free Stock: Video Search",
+        "Free Stock: Image Search",
+        "Multilingual Text to Speech(TTS)",
+        "Remove image background",
+        "Remove Video background",
+        "Video Compositing",
+        "Image Compositing"
+    ]
+
     for i in range(0, len(image_files), 3):
         cols = st.columns(3)
-        for col, img_file, img_text in zip(cols, image_files[i:i+3], image_texts[i:i+3]):
-            with col:
-                st.image(img_file, use_column_width=True)
-                # Markdown을 사용하여 이미지 아래에 굵은 텍스트 추가
-                col.markdown(f"<div style='text-align: center; font-weight: bold; font-size: 20px;'>{img_text}</div>", unsafe_allow_html=True)    
+        for idx, col in enumerate(cols):
+            if i + idx < len(image_files):
+                col.image(image_files[i + idx], use_column_width=True)
+                col.markdown(f"<div style='text-align: center; font-weight: bold; font-size: 20px;'>{image_texts[i + idx]}</div>", unsafe_allow_html=True)
+
+
+
 
 
 elif selected_menu == "Free Stock":