Update app.py

app.py

@@ -1,13 +1,15 @@
 import streamlit as st
 from pyvis.network import Network
 import random
-import streamlit.components.v1 as components
+from streamlit.components.v1 import components
 
 # 애플리케이션 상태 관리
 if "page" not in st.session_state:
     st.session_state.page = "login"  # 초기 페이지는 로그인
-if "names" not in st.session_state:
-    st.session_state.names = []
+if "influencers" not in st.session_state:
+    st.session_state.influencers = []
+if "products" not in st.session_state:
+    st.session_state.products = []
 
 
 # 페이지 전환 함수
@@ -26,24 +28,32 @@ def login_page():
     if st.button("로그인"):
         if username == "admin" and password == "password123":  # 간단한 로그인 검증
             st.success("로그인 성공!")
-            switch_page("name_input")  # 이름 입력 페이지로 이동
+            switch_page("input_data")  # 입력 페이지로 이동
         else:
             st.error("로그인 실패! 사용자 이름 또는 비밀번호가 잘못되었습니다.")
 
 
-# 이름 입력 페이지
-def name_input_page():
-    st.title("이름 입력 페이지")
-    st.write("이름을 한 줄에 하나씩 입력하세요.")
+# 데이터 입력 페이지
+def input_data_page():
+    st.title("데이터 입력 페이지")
     
-    names_input = st.text_area("이름 목록", height=200)
+    # 인플루언서 입력
+    st.subheader("인플루언서 입력")
+    influencers_input = st.text_area("인플루언서 목록 (한 줄에 하나씩 입력하세요)", height=150)
+    
+    # 상품 입력
+    st.subheader("상품 입력")
+    products_input = st.text_area("상품 목록 (한 줄에 하나씩 입력하세요)", height=150)
     
     if st.button("매칭 시작하기"):
-        names = [name.strip() for name in names_input.strip().split('\n') if name.strip()]
-        if len(names) < 2:
-            st.error("매칭할 이름이 2개 이상 필요합니다.")
+        influencers = [name.strip() for name in influencers_input.strip().split('\n') if name.strip()]
+        products = [name.strip() for name in products_input.strip().split('\n') if name.strip()]
+        
+        if not influencers or not products:
+            st.error("인플루언서와 상품을 모두 입력해야 합니다.")
         else:
-            st.session_state.names = names
+            st.session_state.influencers = influencers
+            st.session_state.products = products
             switch_page("match_result")  # 매칭 결과 페이지로 이동
 
 
@@ -51,48 +61,46 @@ def name_input_page():
 def match_result_page():
     st.title("매칭 결과 페이지")
     
-    if not st.session_state.names:
-        st.warning("이름 입력 페이지에서 이름을 먼저 입력하세요!")
+    if not st.session_state.influencers or not st.session_state.products:
+        st.warning("데이터 입력 페이지에서 인플루언서와 상품을 먼저 입력하세요!")
         return
     
-    names = st.session_state.names
-    random.shuffle(names)
-    mid = len(names) // 2
-    group_A = names[:mid]
-    group_B = names[mid:]
-    
-    # 매칭 로직
-    matching_pairs = [(a, b) for a, b in zip(group_A, group_B)]
-    unmatched = set(names) - set(a for a, _ in matching_pairs) - set(b for _, b in matching_pairs)
+    influencers = st.session_state.influencers
+    products = st.session_state.products
+
+    # 매칭 로직: 랜덤으로 여러 인플루언서를 하나의 상품에 연결
+    matches = []
+    for influencer in influencers:
+        product = random.choice(products)  # 상품 중 랜덤으로 선택
+        matches.append((influencer, product))
     
     # pyvis 네트워크 생성
     net = Network(height='600px', width='100%', bgcolor='#222222', font_color='white')
     
-    for name in names:
-        net.add_node(name, label=name)
+    # 노드 추가
+    for influencer in influencers:
+        net.add_node(influencer, label=influencer, color="blue")
+    for product in products:
+        net.add_node(product, label=product, color="green")
     
-    for a, b in matching_pairs:
-        net.add_edge(a, b, color="red", width=3)
+    # 매칭 간선 추가
+    for influencer, product in matches:
+        net.add_edge(influencer, product, color="red", width=2)
     
     # HTML 렌더링
     net_html = net.generate_html()
-    components.html(net_html, height=600)  # 수정된 부분
+    components.html(net_html, height=600)
 
     # 매칭 결과 출력
     st.subheader("매칭 결과:")
-    for pair in matching_pairs:
-        st.write(f"{pair[0]} - {pair[1]}")
-    
-    if unmatched:
-        st.subheader("매칭되지 않은 사람:")
-        for name in unmatched:
-            st.write(name)
+    for influencer, product in matches:
+        st.write(f"{influencer} ➡️ {product}")
 
 
 # 페이지 라우팅
 if st.session_state.page == "login":
     login_page()
-elif st.session_state.page == "name_input":
-    name_input_page()
+elif st.session_state.page == "input_data":
+    input_data_page()
 elif st.session_state.page == "match_result":
     match_result_page()