refactor: disconnect DB module

app.py

@@ -12,7 +12,7 @@ from audiorecorder import audiorecorder
 
 # internal modules
 from bots.judgement_bot import debate_judgement
-from modules.db_modules import get_db, put_item, get_all_items, get_lastest_item
+# from modules.db_modules import get_db, put_item, get_lastest_item
 from modules.gpt_modules import gpt_call, gpt_call_context
 from modules.whisper_modules import whisper_transcribe
 from modules.setting_modules import blockPrint
@@ -30,7 +30,7 @@ st.set_page_config(page_title="Debate With GPT : DEBO")
 #########################################################
 # GET DB
 #########################################################
-dynamodb = get_db()
+# dynamodb = get_db()
 
 #########################################################
 # Time Stamp
@@ -112,18 +112,12 @@ def page_1_2_controller():
 def page_2_4_controller():
     st.session_state.page = "Page 4"
 
-# def page_3_4_controller():
-#     st.session_state.page = "Page 4"
-
 def page_4_5_controller():
     st.session_state.page = "Page 5"
 
 def page_5_6_controller():
     st.session_state.page = "Page 6"
 
-# def page_2_7_controller():
-#     st.session_state.page = "Page 7"
-
 def page_n_1_controller():
     st.session_state.page = "Page 1"
 
@@ -161,18 +155,18 @@ def save_info(user_id):
     #########################################################
     # Session Update
     #########################################################
-    debate_setting = get_lastest_item(
-        table=dynamodb.Table('DEBO_debate_setting'),
-        name_of_partition_key="user_id",
-        value_of_partition_key=st.session_state.user_id,
-        limit_num=1
-    )
-    # Session이 없다면, 0으로 초기화
-    if not debate_setting:
-        st.session_state.session_num = 0
-    # User의 이전 기록에서 Session이 있다면, Session Number를 가져오고 갱신함
-    else:
-        st.session_state.session_num = debate_setting[0]['session_num']
+    # debate_setting = get_lastest_item(
+    #     table=dynamodb.Table('DEBO_debate_setting'),
+    #     name_of_partition_key="user_id",
+    #     value_of_partition_key=st.session_state.user_id,
+    #     limit_num=1
+    # )
+    # if not debate_setting:
+    #     st.session_state.session_num = 0
+    # else:
+    #     st.session_state.session_num = debate_setting[0]['session_num']
+    st.session_state.session_num = 0
+
 
 def page1():
     val_id = False
@@ -260,51 +254,6 @@ def page2():
         on_click=page_control_func,
     )
 
-#########################################################
-# Page 3
-#########################################################
-# def page3():
-#     debate_history = get_lastest_item(
-#             table=dynamodb.Table('DEBO_debate_setting'),
-#             name_of_partition_key="user_id",
-#             value_of_partition_key=st.session_state.user_id,
-#             #TODO 전체 보여줄 개수 설정
-#             limit_num=10
-#         )
-
-#     if not debate_history:
-#         st.info('There is no previous debate history', icon="ℹ️")
-
-#     _, _, pre, home  = st.columns([5, 5, 1, 1])
-#     with pre:
-#         st.button("🔙", on_click=page_1_2_controller, use_container_width=True)
-#     with home:
-#         st.button("🔝", on_click=page_n_1_controller, use_container_width=True)
-#     st.header("Debate History")
-
-#     st.button(
-#         label=f'🚀 Start new debate',
-#         type='primary',
-#         on_click=page_3_4_controller
-#     )
-#     st.write("_"*50)
-
-#     num_history = len(debate_history)
-#     for i in range(num_history):
-#         with st.container():
-#             st.write(f"#### {i + 1}")
-#             st.write(f"Debate Thema : {debate_history[i]['debate_theme']}")
-#             st.write(f"Debate Topic : {debate_history[i]['debate_topic']}")
-#             st.write(f"Case 1 : {debate_history[i]['case1']}")
-#             st.write(f"Case 2 : {debate_history[i]['case2']}")
-#             st.write(f"Case 3 : {debate_history[i]['case3']}")
-#             st.write(f"Created at : {debate_history[i]['time_stamp']}")
-#             st.button(
-#                 label='Continue this dabate',
-#                 key=str(i),
-#                 on_click=page_4_5_controller
-#             )
-#             st.write("_"*50)
 
 #########################################################
 # Page 4
@@ -315,19 +264,19 @@ def store_debate_data(checked, case1, case2, case3):
     if not checked:
         st.session_state.case1, st.session_state.case2, st.session_state.case3 = case1, case2, case3
 
-    put_item(
-        table=dynamodb.Table('DEBO_debate_setting'),
-        item={
-            'user_id': st.session_state.user_id,
-            'time_stamp': time_stamp,
-            'debate_theme': st.session_state.debate_theme,
-            'debate_topic': st.session_state.topic,
-            'case1': st.session_state.case1,
-            'case2': st.session_state.case2,
-            'case3': st.session_state.case3,
-            'session_num': st.session_state.session_num,
-            }
-    )
+    # put_item(
+    #     table=dynamodb.Table('DEBO_debate_setting'),
+    #     item={
+    #         'user_id': st.session_state.user_id,
+    #         'time_stamp': time_stamp,
+    #         'debate_theme': st.session_state.debate_theme,
+    #         'debate_topic': st.session_state.topic,
+    #         'case1': st.session_state.case1,
+    #         'case2': st.session_state.case2,
+    #         'case3': st.session_state.case3,
+    #         'session_num': st.session_state.session_num,
+    #         }
+    # )
 
 def page4():
     #########################################################
@@ -423,7 +372,7 @@ def page4():
         key="disabled",
     )
     #########################################################
-    # Case도 세션에 저장
+    # Save case in session
     #########################################################
     case1 = st.text_area(
         label="Write a Case 1",
@@ -498,17 +447,17 @@ def page4():
                         st.experimental_rerun()
                     st.stop()
 
-                # save user_prompt and bot_response to database
-                put_item(
-                    table=dynamodb.Table('DEBO_gpt_ask'),
-                    item={
-                        'user_id': st.session_state.user_id,
-                        'time_stamp': time_stamp,
-                        'user_prompt': user_input,
-                        'bot_response': result,
-                        'session_num': st.session_state.session_num,
-                    }
-                )
+                # Save user_prompt and bot_response to database
+                # put_item(
+                #     table=dynamodb.Table('DEBO_gpt_ask'),
+                #     item={
+                #         'user_id': st.session_state.user_id,
+                #         'time_stamp': time_stamp,
+                #         'user_prompt': user_input,
+                #         'bot_response': result,
+                #         'session_num': st.session_state.session_num,
+                #     }
+                # )
 
         else:
             result = st.session_state.ask_gpt_prev_response
@@ -585,16 +534,16 @@ def page5():
                         st.experimental_rerun()
                     st.stop()
                 
-                put_item(
-                    table=dynamodb.Table('DEBO_gpt_ask'),
-                    item={
-                        'user_id': st.session_state.user_id,
-                        'time_stamp': time_stamp,
-                        'user_prompt': user_input,
-                        'bot_response': result,
-                        'session_num': st.session_state.session_num,
-                    }
-                )
+                # put_item(
+                #     table=dynamodb.Table('DEBO_gpt_ask'),
+                #     item={
+                #         'user_id': st.session_state.user_id,
+                #         'time_stamp': time_stamp,
+                #         'user_prompt': user_input,
+                #         'bot_response': result,
+                #         'session_num': st.session_state.session_num,
+                #     }
+                # )
         else:
             result = st.session_state.ask_gpt_prev_response
 
@@ -693,28 +642,28 @@ def page5():
                     st.stop()
                 st.session_state['pre_audio'] = audio
 
-                debate_main_latest_data = get_lastest_item(
-                    table=dynamodb.Table('DEBO_debate_main'),
-                    name_of_partition_key="user_id",
-                    value_of_partition_key=st.session_state.user_id,
-                    limit_num=1
-                )
-                if not debate_main_latest_data:
-                    turn_num = 0
-                else:
-                    turn_num = debate_main_latest_data[0]['turn_num']
-
-                put_item(
-                    table=dynamodb.Table('DEBO_debate_main'),
-                    item={
-                        'user_id': st.session_state.user_id,
-                        'time_stamp': time_stamp,
-                        'session_num': st.session_state.session_num,
-                        'bot_response': response,
-                        'user_prompt': user_input,
-                        'turn_num': turn_num,
-                    }
-                )
+                # debate_main_latest_data = get_lastest_item(
+                #     table=dynamodb.Table('DEBO_debate_main'),
+                #     name_of_partition_key="user_id",
+                #     value_of_partition_key=st.session_state.user_id,
+                #     limit_num=1
+                # )
+                # if not debate_main_latest_data:
+                #     turn_num = 0
+                # else:
+                #     turn_num = debate_main_latest_data[0]['turn_num']
+
+                # put_item(
+                #     table=dynamodb.Table('DEBO_debate_main'),
+                #     item={
+                #         'user_id': st.session_state.user_id,
+                #         'time_stamp': time_stamp,
+                #         'session_num': st.session_state.session_num,
+                #         'bot_response': response,
+                #         'user_prompt': user_input,
+                #         'turn_num': turn_num,
+                #     }
+                # )
             else:
                 send_error_message.error("Please record your voice first", icon="🚨")
                 reload = True
@@ -739,7 +688,6 @@ def page5():
         audio_bytes = audio_file.read()
         st.audio(audio_bytes, format='audio/ogg')
 
-        #TODO zip_longest() 고려하기!
         message_pairs = zip(
             st.session_state['bot_debate_history'][1:],
             st.session_state['user_debate_history'],
@@ -747,8 +695,7 @@ def page5():
         for i, (bot_hist, user_hist) in enumerate(message_pairs):
             message(user_hist, is_user=True, key=str(i)+'_user')
             message(bot_hist, key=str(i + 1)+'_bot')
-            # if bot_hist:
-            #TODO 생성된 message와 음성 파일 path를 하나의 객체로 관리하는 방법 고민
+
             if i == len(st.session_state['bot_debate_history']) - 2 and not reload:
                 text_to_speech = gTTS(text=bot_hist, lang='en', slow=False)
                 text_to_speech.save(f"audio/ses_{st.session_state['session_num']}_bot_res_{str(i + 1)}.mp3")
@@ -808,9 +755,6 @@ def page6():
     with tab1:
         st.header("Debate Evaluation")
 
-        # 전체, 유저, 봇 세 가지 옵션 중에 선택
-        #judgement_who = st.selectbox("Choose what you want! (Evaluation result / Perfect case on this theme)", debate_themes)
-
         if st.session_state.judgement_result == "":
             with st.spinner('Wait for result...'):
                 judgement_result = ""
@@ -830,40 +774,20 @@ def page6():
                 st.write("Debate Judgement Result")
                 st.write(judgement_result)
 
-                if judgement_result != "":
-                    put_item(
-                        table=dynamodb.Table('DEBO_evaluation'),
-                        item={
-                            'user_id': st.session_state.user_id,
-                            'time_stamp': time_stamp,
-                            'judgement_text': judgement_result,
-                            'session_num': st.session_state.session_num,
-                        }
-                    )
+                # if judgement_result != "":
+                #     put_item(
+                #         table=dynamodb.Table('DEBO_evaluation'),
+                #         item={
+                #             'user_id': st.session_state.user_id,
+                #             'time_stamp': time_stamp,
+                #             'judgement_text': judgement_result,
+                #             'session_num': st.session_state.session_num,
+                #         }
+                #     )
             st.success('Done!')
         else:
             st.write(st.session_state.judgement_result)
 
-    # with tab2:
-    #     st.header("Perfect Case")
-
-    #     perfect_case_list = [
-    #         "This house supports the creation of an international court with a mandate to prosecute leaders for health crimes",
-    #         "This house believes that governments would be justified in heavily pursuing long-termism",
-    #         "THBT international discussion forums should not self-censor* in an attempt to increase inclusivity to people from countries with stringent freedom-of-speech rules.",
-    #         ]
-
-    #     perfect_case_selected = st.selectbox("Choose the Perfect Case", perfect_case_list)
-
-    #     perfect_case = perfect_case_selector(
-    #         perfect_case_selected
-    #         )
-        
-    #     perfect_case_url = perfect_case['perfect_case_url']
-    #     perfect_case_text = perfect_case['perfect_case_text']
-
-    #     st.video(perfect_case_url)
-    #     st.write(perfect_case_text)
 
     with tab2:
         st.header('Debate Analysis')
@@ -910,153 +834,19 @@ def page6():
         disfluency_counts = sum(user_word in disfluency_word_list for user_word in user_history)
         st.write("Disfluency Counts: ", disfluency_counts)
 
-        if total_word_count != "" and average_word_per_time != "" and disfluency_counts != "":
-
-                put_item(
-                    table=dynamodb.Table('DEBO_debate_analysis'),
-                    item={
-                        'user_id': st.session_state.user_id,
-                        'time_stamp': time_stamp,
-                        'total_word_count': total_word_count,
-                        'average_word_per_time': Decimal(str(average_word_per_time)),
-                        'disfluency_counts': disfluency_counts,
-                        'session_num': int(st.session_state.session_num),
-                    }
-                )
-
-        # 유저와 봇의 대화 데이터가 세션에 남아있음
-        # st.session_state.debate_history
-    
-    ############################################
-    # Visualization
-    ############################################
-
-    # 이전에 기록된 값이 있다면, 그래프를 그립니다.
-    # ���전에 기록된 값이 없다면, 그래프를 그리지 않습니다.
-
-
-#########################################################
-# Page7
-#########################################################
-
-# def page7():
-
-#     # end time
-#     st.session_state.end_time = time.time()
-#     st.session_state.debate_time = st.session_state.end_time - st.session_state.start_time
-
-#     _, _, pre, home  = st.columns([5, 5, 1, 1])
-#     with pre:   
-#         st.button("🔙", on_click=page_1_2_controller, use_container_width=True)
-#     with home:
-#         st.button("🔝", on_click=page_n_1_controller, use_container_width=True)
-
-#     st.header('Total Debate Evaluation')
+        # if total_word_count != "" and average_word_per_time != "" and disfluency_counts != "":
 
-#     tab1, tab2 = st.tabs(['Debate Judgement', 'Debate Analysis'])
-
-#     with tab1:
-#         st.header("Debate Evaluation")
-        
-#         debate_themes = ['User-Bot', "User", "Bot"]
-
-#         # 전체, 유저, 봇 세 가지 옵션 중에 선택
-#         judgement_who = st.selectbox("Choose your debate theme", debate_themes)
-
-#         if st.session_state.judgement_result == "":
-#             with st.spinner('Wait for judgement result...'):
-#                 judgement_result = ""
-
-#                 user_debate_history = "".join(
-#                     st.session_state.user_debate_history
-#                 )
-#                 bot_debate_history = "".join(
-#                     st.session_state.bot_debate_history
-#                 )
-
-#                 judgement_result = debate_judgement(
-#                     judgement_who, 
-#                     user_debate_history, 
-#                     bot_debate_history
-#                     )
-                
-#                 st.session_state.judgement_result = judgement_result
-                
-#                 st.write("Debate Judgement Result")
-#                 st.write(judgement_result)
-
-#                 if judgement_result:
-#                         put_item(
-#                             table=dynamodb.Table('DEBO_evaluation'),
-#                             item={
-#                                 'user_id': st.session_state.user_id,
-#                                 'time_stamp': time_stamp,
-#                                 'judgement_text': judgement_result,
-#                                 'session_num': int(st.session_state.session_num),
-#                             }
-#                         )
-#                 st.success('Done!')
-#         else:
-#             st.write(st.session_state.judgement_result)
-
-#     with tab2:
-#         st.header('Debate Analysis')
-
-#         # 유저의 history를 기반으로 발화량, 빈출 단어, 발화 습관 세 가지를 분석
-#         user_history = st.session_state.user_debate_history
-
-#         # 1. 발화량: 총 단어, 평균 속도(단어/시간)를 평균 발화량 혹은 참고 지표와 비교해 제시
-
-#         # 총 단어
-#         # 텍스트를 단어로 분할합니다.
-#         # 각 단어의 빈도를 계산합니다.
-#         total_word_count = len(user_history)
-#         #total_word_count = len(user_history.split())
-#         st.write("Total Word Count: ", total_word_count)
-
-#         # 평균 속도(단어/시간)
-#         #user_debate_time = st.session_state.user_debate_time
-#         average_word_per_time = total_word_count / st.session_state.debate_time # 시간 단위보고 나중에 수정하기
-#         st.write("Average Word Per Time: ", average_word_per_time)
-
-#         # 2. 빈출 단어: 반복해서 사용하는 단어 리스트
-#         # 빈도 계산
-#         frequency = Counter(user_history)
-#         # 가장 빈도가 높은 데이터 출력
-#         most_common_data = frequency.most_common(10)
-#
-#         print(most_common_data)
-#         st.write("Most Common Words: ", most_common_data)
-
-#         # 3. 발화 습관: 불필요한 언어습관(아, 음)
-#         # whisper preprocesser에서 주면
-#         disfluency_word_list = ['eh', 'umm', 'ah', 'uh', 'er', 'erm', 'err']
-#         # Count the disfluency words
-#         disfluency_counts = sum(user_word in disfluency_word_list for user_word in user_history)
-#         st.write("Disfluency Counts: ", disfluency_counts)
-
-#         if total_word_count != "" and average_word_per_time != "" and disfluency_counts != "":
-#                 put_item(
-#                     table=dynamodb.Table('DEBO_evaluation'),
-#                     item={
-#                         'user_id': st.session_state.user_id,
-#                         'time_stamp': time_stamp,
-#                         'total_word_count': total_word_count,
-#                         'average_word_per_time': Decimal(str(average_word_per_time)),
-#                         'disfluency_counts': disfluency_counts,
-#                         'session_num': int(st.session_state.session_num),
-#                     }
-#                 )
-
-        # 유저와 봇의 대화 데이터가 세션에 남아있음
-        # st.session_state.debate_history
-
-    ############################################
-    # Visualization
-    ############################################
-
-    # 이전에 기록된 값이 있다면, 그래프를 그립니다.
-    # 이전에 기록된 값이 없다면, 그래프를 그리지 않습니다.
+        #         put_item(
+        #             table=dynamodb.Table('DEBO_debate_analysis'),
+        #             item={
+        #                 'user_id': st.session_state.user_id,
+        #                 'time_stamp': time_stamp,
+        #                 'total_word_count': total_word_count,
+        #                 'average_word_per_time': Decimal(str(average_word_per_time)),
+        #                 'disfluency_counts': disfluency_counts,
+        #                 'session_num': int(st.session_state.session_num),
+        #             }
+        #         )
 
 
 #########################################################
@@ -1065,11 +855,9 @@ def page6():
 pages = {
     "Page 1": page1, # user_id를 입력받는 페이지
     "Page 2": page2, # 원하는 기능을 선택하는 페이지
-    # "Page 3": page3, # 과거 토론 내역을 선택하는 페이지
     "Page 4": page4, # 토론 세부사항 설정
     "Page 5": page5, # Total Debate
     "Page 6": page6, # Evaluation Only
-    # "Page 7": page7, # Analyzing Utterances
 }
 
 selection = st.session_state.page