Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,14 +1,17 @@
|
|
1 |
import streamlit as st
|
2 |
import os
|
3 |
from dotenv import load_dotenv
|
4 |
-
|
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 |
-
#
|
|
|
|
|
|
|
|
|
|
|
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()
|