Jiaaaaaaax commited on
Commit
66cd554
·
verified ·
1 Parent(s): f6ec2df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -6
app.py CHANGED
@@ -1,14 +1,17 @@
1
  import streamlit as st
2
  import os
3
  from dotenv import load_dotenv
4
- from audio_recorder_streamlit import audio_recorder
5
- import time
6
  from pathlib import Path
7
  import json
8
 
9
  # Load environment variables
10
  load_dotenv()
11
 
 
 
 
 
12
  # Page config
13
  st.set_page_config(
14
  page_title="MI Assistant",
@@ -23,7 +26,12 @@ if 'role' not in st.session_state:
23
  if 'current_mode' not in st.session_state:
24
  st.session_state.current_mode = None
25
 
26
- # Sidebar
 
 
 
 
 
27
  def sidebar():
28
  with st.sidebar:
29
  # Role switcher at the top
@@ -48,7 +56,6 @@ def sidebar():
48
  if st.button("Session Analysis"):
49
  st.session_state.current_mode = "session_analysis"
50
 
51
- # Welcome page
52
  def welcome_page():
53
  st.title("Welcome to MI Assistant")
54
  st.write("Please select your role to continue:")
@@ -63,7 +70,6 @@ def welcome_page():
63
  st.session_state.role = "Therapist"
64
  st.session_state.current_mode = "tutorial"
65
 
66
- # Main app logic
67
  def main():
68
  if st.session_state.role is None:
69
  welcome_page()
@@ -80,4 +86,4 @@ def main():
80
  show_session_analysis()
81
 
82
  if __name__ == "__main__":
83
- main()
 
1
  import streamlit as st
2
  import os
3
  from dotenv import load_dotenv
4
+ import google.generativeai as genai
 
5
  from pathlib import Path
6
  import json
7
 
8
  # Load environment variables
9
  load_dotenv()
10
 
11
+ # Configure Gemini API
12
+ genai.configure(api_key=os.getenv("Gemini_API_Key"))
13
+ model = genai.GenerativeModel('gemini-pro')
14
+
15
  # Page config
16
  st.set_page_config(
17
  page_title="MI Assistant",
 
26
  if 'current_mode' not in st.session_state:
27
  st.session_state.current_mode = None
28
 
29
+ # Import other modules
30
+ from tutorial import show_tutorial
31
+ from live_session import show_live_session
32
+ from moti_chat import show_moti_chat
33
+ from session_analysis import show_session_analysis
34
+
35
  def sidebar():
36
  with st.sidebar:
37
  # Role switcher at the top
 
56
  if st.button("Session Analysis"):
57
  st.session_state.current_mode = "session_analysis"
58
 
 
59
  def welcome_page():
60
  st.title("Welcome to MI Assistant")
61
  st.write("Please select your role to continue:")
 
70
  st.session_state.role = "Therapist"
71
  st.session_state.current_mode = "tutorial"
72
 
 
73
  def main():
74
  if st.session_state.role is None:
75
  welcome_page()
 
86
  show_session_analysis()
87
 
88
  if __name__ == "__main__":
89
+ main()