Update app.py

app.py

@@ -29,8 +29,6 @@ model.eval()
 view_flag = True
 skip = False
 
-left_column, right_column = st.columns(2)
-
 def predict_image(img):
     img = img.convert('RGB')
     img_transformed = transform(img)
@@ -41,10 +39,9 @@ def predict_image(img):
     return preds.item()
 uploaded_file = st.file_uploader('Choose an image...', type=['jpg', 'png'])
 if uploaded_file:
-    with left_column:
-        img = Image.open(uploaded_file)
-        st.image(img, caption="Uploaded Image", use_column_width=True)
-        pred = predict_image(img)
+    img = Image.open(uploaded_file)
+    st.image(img, caption="Uploaded Image", use_column_width=True)
+    pred = predict_image(img)
     if pred == 0:
         season_type = "秋"
     elif pred == 1:
@@ -57,35 +54,43 @@ if uploaded_file:
         st.session_state.show_video = False
     if 'skip' not in st.session_state:
         st.session_state.skip = False
+    if 'result' not in st.session_state:
+        st.session_state.result = False
 
     person_result = st.selectbox(
         '結果の修正',
         ('春', '夏', '秋', '冬')
-    )
+)
 
     if person_result:
         season_type = person_result
 
-    with left_column:
-        st.write(f"パーソナルカラーは {season_type} です")
+    st.write(f"パーソナルカラーは {season_type} です")
+
+    # おすすめの商品を見るボタン
+    view_result = st.button("診断結果詳細 & おすすめ商品を見る")
+    if view_result or st.session_state.result:
+        st.session_state.result = True
+        st.image(f"{season_type}.png")
 
-        # おすすめの商品を見るボタン
+    if st.session_state.result and not st.session_state.show_video:
         view_recommend = st.button("おすすめの商品を見る")
-    with right_column:
         if view_recommend:
             st.session_state.show_video = True
 
-        # スキップされずに動画を表示する
-        if st.session_state.show_video and not st.session_state.skip:
-            st.video("sample.mp4", start_time=0)
+    # スキップされずに動画を表示する
+    if st.session_state.show_video and not st.session_state.skip:
+        st.video("sample.mp4", start_time=0)
 
-        # 動画のスキップボタンが押されたら、レコメンド文を表示する
-        if st.session_state.show_video and not st.session_state.skip:
-            time.sleep(5)
-            st.session_state.show_video = False
-            st.session_state.skip = True
-        if st.session_state.skip:
-            st.write(
-                """
-                レコメンド文です。
-                """)
+    # 動画のスキップボタンが押されたら、レコメンド文を表示する
+    if st.session_state.show_video and not st.session_state.skip:
+        time.sleep(5)
+        st.session_state.show_video = False
+        st.session_state.skip = True
+    if st.session_state.skip:
+        st.write(
+            """
+            おすすめの商品です。
+            """)
+        st.image("服.png")
+