Revert "refactor: disconnect DynomoDB module"

This reverts commit 3bcbe7ac72c79a1b4945d2c75f7898dccb071b5b.

app.py

@@ -4,6 +4,7 @@ import pprint
 import time
 import openai
 
+from decimal import Decimal
 from gtts import gTTS
 from collections import Counter
 from streamlit_chat import message
@@ -11,6 +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.gpt_modules import gpt_call, gpt_call_context
 from modules.whisper_modules import whisper_transcribe
 from modules.setting_modules import blockPrint
@@ -25,6 +27,11 @@ blockPrint()
 #########################################################
 st.set_page_config(page_title="Debate With GPT : DEBO")
 
+#########################################################
+# GET DB
+#########################################################
+dynamodb = get_db()
+
 #########################################################
 # Time Stamp
 #########################################################
@@ -154,10 +161,18 @@ def save_info(user_id):
     #########################################################
     # Session Update
     #########################################################
-
-    # Session 무조건 0으로 초기화
-    st.session_state.session_num = 0
-
+    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']
 
 def page1():
     val_id = False
@@ -300,6 +315,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,
+            }
+    )
 
 def page4():
     #########################################################
@@ -470,6 +498,18 @@ 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,
+                    }
+                )
+
         else:
             result = st.session_state.ask_gpt_prev_response
 
@@ -545,6 +585,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,
+                    }
+                )
         else:
             result = st.session_state.ask_gpt_prev_response
 
@@ -642,8 +692,29 @@ def page5():
                         st.experimental_rerun()
                     st.stop()
                 st.session_state['pre_audio'] = audio
-                turn_num = 0
 
+                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
@@ -759,6 +830,16 @@ 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,
+                        }
+                    )
             st.success('Done!')
         else:
             st.write(st.session_state.judgement_result)
@@ -829,6 +910,20 @@ 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