Merge remote-tracking branch 'upstream/master' into prototype_v0.1

.github/workflows/cd.yml

@@ -0,0 +1,23 @@
+name: Sync to Hugging Face hub
+
+on:
+  push:
+    branches:
+      - master
+    paths-ignore: ["/README.md"]
+
+  workflow_dispatch:
+
+jobs:
+  sync-to-hub:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+          lfs: true
+      - name: Push to hub
+        env:
+          HF_TOKEN: ${{ secrets.HF_TOKEN }}
+        run: git push https://Ridealist:[email protected]/spaces/Ridealist/chat_bot main
+          

app.py

@@ -8,6 +8,7 @@ from collections import Counter
 from streamlit_chat import message
 
 from bots.judgement_bot import debate_judgement
+from bots.perfect_case_bot import perfect_case_selector
 import time
 from time import strftime
 
@@ -93,6 +94,9 @@ if "pre_audio" not in st.session_state:
 if "session_num" not in st.session_state:
     st.session_state.session_num = 0
 
+if "judgement_result" not in st.session_state:
+    st.session_state.judgement_result = ""
+
 
 #########################################################
 # Save function (placeholder)
@@ -161,9 +165,11 @@ def page_n_1_controller():
 # Page 1
 #########################################################
 def page1():
-    _, _, _, home  = st.columns([5, 5, 1, 1])
-    with home:
-        st.button("🔝", on_click=page_n_1_controller, use_container_width=True)
+
+    # 첫페이지는 home이 따로 필요없을 것 같아서 주석처리함
+    #_, _, _, home  = st.columns([5, 5, 1, 1])
+    # with home:
+    #     st.button("🔝", on_click=page_n_1_controller, use_container_width=True)
 
     st.header('User Info')
     st.session_state.user_id = st.text_input(
@@ -451,24 +457,39 @@ def page4():
 #########################################################
 
 def generate_response(prompt):
+
     st.session_state['user_debate_history'].append(prompt)
     st.session_state['total_debate_history'].append({"role": "user", "content": prompt})
-    response = gpt_call_context(st.session_state['total_debate_history'])
+
+    if len(prompt.split()) < 10:
+        response = "Please speak longer!"
+    else:
+        response = gpt_call_context(st.session_state['total_debate_history'])
+    
     st.session_state['bot_debate_history'].append(response)
     st.session_state['total_debate_history'].append({"role": "assistant", "content": response})
     return response
 
 def execute_stt(audio, error_message):
-    wav_file = open("audio/audio.wav", "wb")
+
+    # audio 기록 누적
+    #user_audio_path = "audio/" + str(st.session_state.user_id) + "_" + str(st.session_state.session_num) + "_" + str(time.time()) + ".wav"
+    # audio 기록을 누적하고 싶지 않다면
+    user_audio_path = "audio/audio.wav"
+
+    wav_file = open(user_audio_path, "wb")
     wav_file.write(audio.tobytes())
-    wav_file.close()
+
     try:
-        user_input = whisper_transcribe(audio)
+        user_input = whisper_transcribe(wav_file)
     except:
         error_message.error("Whisper Error : The engine is currently overloaded, it will be auto-reloaded in a second")
         time.sleep(1)
         st.experimental_rerun()
 
+    # close file
+    wav_file.close()
+
     return user_input
 
 def page5():
@@ -583,8 +604,10 @@ def page5():
         if submit_buttom:
             if audio.any():
                 user_input = execute_stt(audio, openai_error_bottom)
+
                 try :
                     response = generate_response(user_input)
+                    print("response", response)
                 except:
                     openai_error_bottom.error("Chat-GPT Error : The engine is currently overloaded, it will be auto-reloaded in a second")
                     time.sleep(1)
@@ -670,48 +693,59 @@ def page6():
 
     st.write('Note that evaluation using GPT is an experimental feature. Please check it out and give us your feedback.')
 
-    tab1, tab2 = st.tabs(['Debate Evaluation', 'Debate Analysis'])
+    tab1, tab2, tab3 = st.tabs(['Debate Evaluation', 'Perfect Case', 'Debate Analysis'])
 
     with tab1:
         st.header("Debate Evaluation")
-        
-        debate_themes = ['User-Bot', "User", "Bot"]
 
         # 전체, 유저, 봇 세 가지 옵션 중에 선택
-        judgement_who = st.selectbox("Choose your debate theme", debate_themes)
+        #judgement_who = st.selectbox("Choose what you want! (Evaluation result / Perfect case on this theme)", debate_themes)
 
-        with st.spinner('Wait for judgement result...'):
-            judgement_result = ""
+        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
+                user_debate_history = "".join(
+                    st.session_state.user_debate_history
+                )
+                bot_debate_history = "".join(
+                    st.session_state.bot_debate_history
                 )
 
-            st.write("Debate Judgement Result")
-            st.write(judgement_result)
+                judgement_result = debate_judgement(
+                    user_debate_history, 
+                    bot_debate_history
+                    )
 
-            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!')
+                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)
 
     with tab2:
+        st.header("Perfect Case")
+
+        perfect_case = perfect_case_selector(
+            st.session_state.debate_theme, 
+            st.session_state.topic
+            )
+
+        st.write(perfect_case)
+
+
+    with tab3:
         st.header('Debate Analysis')
 
         # 유저의 history를 기반으로 발화량, 빈출 단어, 발화 습관 세 가지를 분석
@@ -722,7 +756,10 @@ def page6():
         # 총 단어
         # 텍스트를 단어로 분할합니다.
         # 각 단어의 빈도를 계산합니다.
-        total_word_count = len(user_history)
+
+        # 리스트를 문자열로 변환하고, 공백을 기준으로 단어를 분할합니다.
+        total_word_list = "".join(user_history).split()
+        total_word_count = len(total_word_list)
         #total_word_count = len(user_history.split())
         st.write("Total Word Count: ", total_word_count)
 
@@ -733,11 +770,16 @@ def page6():
 
         # 2. 빈출 단어: 반복해서 사용하는 단어 리스트
         # 빈도 계산
-        frequency = Counter(user_history)
+        frequency = Counter(total_word_list)
         # 가장 빈도가 높은 데이터 출력
-        most_common_data = frequency.most_common(10)
-        print(most_common_data)
-        st.write("Most Common Words: ", most_common_data)
+        most_common_data = frequency.most_common(5)
+
+        st.write("Most Common Words: ")
+        for word, count in most_common_data:
+            st.write(" - ", word, ":", count)
+
+        # print(most_common_data)
+        # st.write("Most Common Words: ", most_common_data)
 
         # 3. 발화 습관: 불필요한 언어습관(아, 음)
         # whisper preprocesser에서 주면
@@ -747,13 +789,6 @@ def page6():
         st.write("Disfluency Counts: ", disfluency_counts)
 
         if total_word_count != "" and average_word_per_time != "" and disfluency_counts != "":
-                
-                print("user_id", type(st.session_state.user_id))
-                print("time_stamp", type(time_stamp))
-                print("total_word_count", type(total_word_count))
-                print("average_word_per_time", type(average_word_per_time))
-                print("disfluency_counts", type(disfluency_counts))
-                print("session_num", type(st.session_state.session_num))
 
                 put_item(
                     table=dynamodb.Table('DEBO_debate_analysis'),
@@ -799,36 +834,41 @@ def page7():
         # 전체, 유저, 봇 세 가지 옵션 중에 선택
         judgement_who = st.selectbox("Choose your debate theme", debate_themes)
 
-        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
-            )
+        if st.session_state.judgement_result == "":
+            with st.spinner('Wait for judgement result...'):
+                judgement_result = ""
 
-            judgement_result = debate_judgement(
-                judgement_who, 
-                user_debate_history, 
-                bot_debate_history
+                user_debate_history = "".join(
+                    st.session_state.user_debate_history
+                )
+                bot_debate_history = "".join(
+                    st.session_state.bot_debate_history
                 )
-            
-            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),
-                        }
+                judgement_result = debate_judgement(
+                    judgement_who, 
+                    user_debate_history, 
+                    bot_debate_history
                     )
-            st.success('Done!')
+                
+                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')

bots/judgement_bot.py

@@ -2,33 +2,13 @@ from modules.gpt_modules import gpt_call
 from langchain.prompts import PromptTemplate
 
 def debate_judgement(
-        judgement_who, 
         user_debate_history, 
         bot_debate_history
         ):
-
-    if judgement_who == 'User-Bot':
-
-        judgement_prompt_preset = "\n".join([
-            "!!Instruction!",
-            "You are now the judge of this debate. Evaluate the debate according to the rules below.",
-            "Rule 1. Decide between the USER and BOT.",
-            "Rule 2. Summarize the debate as a whole and what each debater said.",
-            "Rule 3. For each debater, explain what was persuasive and what made the differnce between winning and losing.",
-        ])
-
-        judgement_prompt = "\n".join([
-                judgement_prompt_preset,
-                "User: " + user_debate_history,
-                "Bot: " + bot_debate_history,
-                "Judgement must be logical with paragraphs.",
-                "Do not show Rule",
-                "Write judgement below.",
-                "Judgement: "
-                ])
-
-    elif judgement_who == 'User':
-
+    
+    if len(user_debate_history.split()) < 100:
+        bot_response = "Under the 100 words, evaluation is not possible."
+    else:
         judgement_prompt_preset = "\n".join([
             "!!Instruction!",
             "You are now the judge of this debate. Evaluate the debate according to the rules below.",
@@ -45,24 +25,6 @@ def debate_judgement(
                 "Judgement: "
                 ])
 
-    elif judgement_who == 'Bot':
-
-        judgement_prompt_preset = "\n".join([
-            "!!Instruction!",
-            "You are now the judge of this debate. Evaluate the debate according to the rules below.",
-            "Rule 1. Summarize the debate as a whole and each said.",
-            "Rule 2. Explain what was persuasive and what made the differnce between winning and losing.",
-        ])
-
-        judgement_prompt = "\n".join([
-                judgement_prompt_preset,
-                "Bot: " + bot_debate_history,
-                "Judgement must be logical with paragraphs.",
-                "Do not show Rule",
-                "Write judgement below.",
-                "Judgement: "
-                ])
-
-    bot_response = gpt_call(judgement_prompt)
+        bot_response = gpt_call(judgement_prompt)
     
     return bot_response

bots/perfect_case_bot.py

@@ -0,0 +1,66 @@
+
+
+
+def perfect_case_selector(debate_theme, topic):
+
+    if debate_theme == 'Education':
+        topic_list = [
+            "THBT college entrance examinations should accept students only on the basis of their academic performance in secondary education.", 
+            "THS a world where the government gives cash that individuals can use to freely select their academic preference (including but not limited to school of choice, private academies, and tutoring) instead of funding for public education.",
+            "THW abolish all requirements and evaluation criteria in higher education (i.e., attendance, exams, assignments)."
+            ]
+    elif debate_theme == 'Sports':
+        topic_list = [
+            "THBT having star players for team sports do more harm than good to the team.",
+            "THR the emphasis on winning a medal in the Olympics as a core symbol of success.",
+            "THP a world where sports serves purely entertainment purposes even at the expense of fair play."
+            ]
+    elif debate_theme == 'Religion':
+        topic_list = [
+            "THW, as a religious group/leader, cease attempts at increasing the number of believers and instead prioritize boosting loyalty amongst adherents to the religion.",
+            "Assuming feasibility, TH prefers a world where a panel of church leaders would create a universally accepted interpretation of the Bible that the believers would abide by.",
+            "THW aggressively crackdown on megachurches."
+            ]
+    elif debate_theme == 'Justice':
+        topic_list = [
+            "In 2050, AI robots are able to replicate the appearance, conversation, and reaction to emotions of human beings. However, their intelligence still does not allow them to sense emotions and feelings such as pain, happiness, joy, and etc.",
+            "In the case a human destroys the robot beyond repair, THW charge murder instead of property damage.",
+            "THP a world where the criminal justice system’s role is mainly for victim’s vengeance. THW allow prosecutors and victims to veto assigned judges."
+            ]
+    elif debate_theme == 'Pandemic':
+        topic_list = [
+            "During a pandemic, THBT businesses that benefit from the pandemic should be additionally taxed.",
+            "THW nullify the effect of medical patents in cases of medical emergencies.",
+            "THW ban media content that denies the efficacy of the COVID-19 without substantial evidence."
+            ]
+    elif debate_theme == 'Politics':
+        topic_list = [
+            "Info: The Candle Light Will (촛불민심) is a term derived from the symbolic candle-light protests for the impeachment of the late president Park Geun Hye, commonly used to mean the people’s will to fight against corrupt governments. The Moon administration has frequently referred to the Candle Light Will as the driving force behind its election that grants legitimacy to its policies. THR the ‘candle light will’ narrative in the political discourse of South Korea.",
+            "THW impose a cap on the property and income of politicians.",
+            "THW give the youth extra votes."
+            ]
+    elif debate_theme == 'Minority':
+        topic_list = [
+            "Context: A prominent member of the LGBT movement has discovered that a very influential politician helping the LGBT movement has been lying about their sexual orientation as being gay when they are straight. THW disclose this information.",
+            "THBT the LGBTQIA+ movement should denounce the existence of marriage as opposed to fighting for equal marriage rights.",
+            "THBT the LGBTQIA+ movement should condemn the consumption of movies and TV shows that cast straight actors/actresses in non-heterosexual identified roles."
+            ]
+    else:
+        topic_list = [
+            "THW remove all laws that relate to filial responsibilities.",
+            "THW require parents to receive approval from experts in relevant fields before making crucial decisions for their children.",
+            "Assuming it is possible to measure the ‘societal danger’ of the fetus in the future, THBT the state should raise infants that pose high levels of threat.",
+            "THBT any upper limits on prison sentences for particularly heinous crimes should be abolished.",
+            "THW require dating apps to anonymize profile pictures.",
+            "THW adopt a Pass/Fail grading system for students who suffer from mental health problems (e.g. depression, bipolar disorder, etc.).",
+            "THBT South Korean feminist movements should reject feminist icons that are adversarial and embody violence.",
+            "THBT freedom of speech should be considered obsolete.",
+            "THR the narrative that eccentric personalities are essential to create art.",
+            "THW allow parents of severely mentally disabled children to medically impede their children's physical growth.",
+            "THR the emphasis on longevity in relationships.",
+            "Assuming feasibility, THW choose to continuously relive the happiest moment of one’s life."
+            ]
+        
+    perfect_case_result = "perfect_case_result"
+
+    return perfect_case_result