Update app.py

app.py

@@ -36,31 +36,38 @@ def main():
     model = load_trained_model("product_model.pth")
 
     # Step 3: 추천을 위한 사용자 ID 입력
-    user_id = st.text.input("사용자 ID 입력: ")
-    print(f"사용자 ID: {user_id}에게 추천해줄 상품 찾는 중...")
+    if "user_id" not in st.session_state:
+        st.session_state["user_id"] = ""
 
-    try:
-        # Step 4: 사용자와 가장 유사한 anchor 찾기
-        print(f"사용자 ID: {user_id} 와 가장 유사한 anchor 찾는 중...")
-        most_similar_anchor, most_similar_anchor_embedding = find_most_similar_anchor(
-            user_id, model
-        )
-        print(f"가장 유사한 anchor: {most_similar_anchor}")
+    user_id = st.text_input("사용자 ID 입력:", st.session_state["user_id"])
 
-        # Step 5: anchor와 가장 유사한 상품 찾기
-        print("anchor와 가장 유사한 학습 상품 찾는 중...")
-        most_similar_product, most_similar_product_embedding = (
-            find_most_similar_product(most_similar_anchor_embedding, model)
-        )
-        print(f"anchor와 가장 유사한 학습 상품 ID: {most_similar_product}")
+    if user_id:
+        print(f"사용자 ID: {user_id}에게 추천해줄 상품 찾는 중...")
 
-        # Step 6: 쇼핑몰 상품 추천
-        print("추천 쇼핑몰 상품 찾는 중...")
-        recommended_product_id = recommend_shop_product(most_similar_product_embedding)
-        print(f"추천 쇼핑몰 상품 ID: {recommended_product_id}")
+        try:
+            # Step 4: 사용자와 가장 유사한 anchor 찾기
+            print(f"사용자 ID: {user_id} 와 가장 유사한 anchor 찾는 중...")
+            most_similar_anchor, most_similar_anchor_embedding = find_most_similar_anchor(
+                user_id, model
+            )
+            print(f"가장 유사한 anchor: {most_similar_anchor}")
 
-    except Exception as e:
-        print(f"An error occurred during the recommendation process: {e}")
+            # Step 5: anchor와 가장 유사한 상품 찾기
+            print("anchor와 가장 유사한 학습 상품 찾는 중...")
+            most_similar_product, most_similar_product_embedding = (
+                find_most_similar_product(most_similar_anchor_embedding, model)
+            )
+            print(f"anchor와 가장 유사한 학습 상품 ID: {most_similar_product}")
+
+            # Step 6: 쇼핑몰 상품 추천
+            print("추천 쇼핑몰 상품 찾는 중...")
+            recommended_product_id = recommend_shop_product(most_similar_product_embedding)
+            print(f"추천 쇼핑몰 상품 ID: {recommended_product_id}")
+
+        except Exception as e:
+            print(f"An error occurred during the recommendation process: {e}")
+    else:
+        st.write("사용자 ID를 입력해주세요.")
 
 
 if __name__ == "__main__":