page 5 complete

bots/judgement_bot.py

@@ -0,0 +1,27 @@
+from modules.gpt_modules import gpt_call
+from langchain.prompts import PromptTemplate
+
+def debate_judgement(debate_history):
+
+    judgement_prompt = "\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_template = PromptTemplate(
+        input_variables=["prompt"],
+        template="\n".join([
+            debate_history,
+            judgement_prompt,
+            "Judgement: "
+            ])
+    )
+    judgement_bot_prompt = judgement_prompt_template.format(
+            prompt=""
+    )
+    bot_response = gpt_call(judgement_bot_prompt)
+    
+    return bot_response

vocal_app.py

@@ -1,4 +1,7 @@
 import streamlit as st
+from modules.gpt_modules import gpt_call
+from langchain.prompts import PromptTemplate
+from bots.judgement_bot import debate_judgement
 #import SessionState
 
 # Page Configuration
@@ -28,6 +31,15 @@ if "case3" not in st.session_state:
 if "page2_tab" not in st.session_state:
     st.session_state.page2_tab = "tab1"
 
+if "total_debate_history" not in st.session_state:
+    st.session_state.total_debate_history = ""
+
+if "user_debate_history" not in st.session_state:
+    st.session_state.user_debate_history = ""
+
+if "bot_debate_history" not in st.session_state:
+    st.session_state.bot_debate_history = ""
+
 
 # Save function (placeholder)
 def save_info(user_id, openAI_token, debate_theme):
@@ -60,6 +72,10 @@ def page_2_3_controller():
 def page2_tab_controller():
     st.session_state.page2_tab = "tab2"
 
+
+def page_5_6_controller():
+    st.session_state.page = "Page 6"
+
 #########################################################
 # Page 1
 #########################################################
@@ -231,17 +247,55 @@ def page4():
 
 
 #########################################################
-# Page5
+# Page5 - Total Debate Evaluation
 #########################################################
 def page5():
+    st.header('Debate Judgement')
+    # 유저와 봇의 대화 데이터가 세션에 남아있음
+    # st.session_state.debate_history
+
+    debate_themes = ['User-Bot', "User", "Bot"]
+
+    # 전체, 유저, 봇 세 가지 옵션 중에 선택
+    judgement_who = st.selectbox("Choose your debate theme", debate_themes)
+
+    if judgement_who == 'User-Bot':
+        debate_history = st.session_state.total_debate_history
+    elif judgement_who == 'User':
+        debate_history = st.session_state.user_debate_history
+    elif judgement_who == 'Bot':
+        debate_history = st.session_state.bot_debate_history
+
+    judgement_result = debate_judgement(debate_history)
+
+    st.write("Debate Judgement Result")
+    st.write(judgement_result)
+
+    st.button(
+        label='Move to Debate Dashboard',
+        on_click=page_5_6_controller
+        )
+
+#########################################################
+# Page6
+#########################################################
+def page6():
+    pass
+
+
+#########################################################
+# Page7
+#########################################################
+def page7():
+    pass
+
+
+#########################################################
+# Page8
+#########################################################
+def page8():
+    pass
 
-    with st.sidebar:
-        st.sidebar.title('Ask to GPT')
-        st.sidebar.text_area(
-            label="Input text here", 
-            placeholder="Input text here",
-            height=100)
-        st.sidebar.button("Ask")
 
 
 #########################################################
@@ -253,6 +307,9 @@ pages = {
     "Page 3": page3, # Total Debate
     "Page 4": page4, # Evaluation Only
     "Page 5": page5, # Analyzing Utterances
+    "Page 6": page6,
+    "Page 7": page7,
+    "Page 8": page8
 }
 
 selection = st.session_state.page

whisper_app.py

@@ -24,6 +24,10 @@ def debate(audio):
     # user_words
     user_prompt = openai.Audio.transcribe("whisper-1", file).text
 
+    print("**************************************")
+    print("user_audio transcription", user_prompt)
+    print("**************************************")
+
     # 일단 테스트를 위해 고정함
     debate_subject = "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."