Updated app.py

app.py

@@ -19,6 +19,7 @@ client = storage.Client(credentials=creds)
 
 bucket_name = "kapnotes"
 bucket = client.bucket(bucket_name)
+st.set_page_config(page_title="Kap Notes", layout="wide")
 
 def get_client_names():
     blobs = list(bucket.list_blobs(prefix=""))
@@ -28,16 +29,43 @@ def get_client_names():
         client_names.add(client_name)
     return sorted(client_names)
 
+def validate_data(client_name, date, meeting):
+    summary_blob_name = f"{client_name}/{date}/{meeting}/summary.txt"
+    transcription_blob_name = f"{client_name}/{date}/{meeting}/transcription.txt"
+    audio_blob_name = f"{client_name}/{date}/{meeting}/audio.wav"
+    summary_blob = bucket.blob(summary_blob_name)
+    transcription_blob = bucket.blob(transcription_blob_name)
+    audio_blob = bucket.blob(audio_blob_name)
+    return summary_blob.exists() and transcription_blob.exists() and audio_blob.exists()
+
+def get_meetings_for_date(client_name, date):
+    prefix = f"{client_name}/{date}/"
+    blobs = bucket.list_blobs(prefix=prefix)
+    meetings = set()
+    for blob in blobs:
+        parts = blob.name[len(prefix):].split('/')
+        if len(parts) > 1:
+            meetings.add(parts[0])
+    return sorted(meetings)
+
+def get_dates_for_client(client_name):
+    client = storage.Client()
+    bucket_name = "kapnotes"
+    bucket = client.get_bucket(bucket_name)
+    prefix = f"{client_name}/"
+    blobs = bucket.list_blobs(prefix=prefix)
+    folder_names = set()
+    for blob in blobs:
+        parts = blob.name[len(prefix):].split('/')
+        if len(parts) > 1: 
+            folder_names.add(parts[0])
+    return sorted(folder_names)
+
 def login():
     st.markdown("""
         <style>
         .stApp {
-            background: linear-gradient(
-                125deg,
-                #F0F4F8 0%,
-                #D9E4F5 50%,
-                #E5EFF8 100%
-            );
+            background: linear-gradient(125deg,rgb(253, 250, 220) 0%,rgb(214, 245, 255) 50%, #F8F8FF 100%);
             background-size: 200% 200%;
             animation: gradientMove 10s ease infinite;
         }
@@ -54,46 +82,9 @@ def login():
             margin-bottom: 3rem;
             letter-spacing: 2px;
         }
-        .login-form-container {
-            width: 400px;
-            margin: 0 auto;
-            padding: 2rem;
-            background: linear-gradient(
-                135deg,
-                rgba(0, 0, 0, 0.1) 0%,
-                rgba(0, 0, 0, 0.15) 100%
-            );
-            backdrop-filter: blur(15px);
-            border-radius: 24px;
-            box-shadow: 
-                0 8px 32px rgba(31, 38, 135, 0.37),
-                inset -4px -4px 8px rgba(255, 255, 255, 0.1),
-                inset 4px 4px 8px rgba(0, 0, 0, 0.2);
-            border: 1px solid rgba(255, 255, 255, 0.18);
-        }
-        .stSelectbox label, .stDateInput label {
-            color: #333333 !important;
-            font-weight: bold;
-            font-size: 1rem;
-            text-shadow: 1px 1px 2px rgba(0, 0, 0, 0.5);
-        }
-        .stSelectbox > div > div, .stDateInput > div > div {
-            background: #2E3A47;
-            border: 1px solid rgba(255, 255, 255, 0.3);
-            border-radius: 12px;
-            color: #FFFFFF;
-            font-weight: bold;
-            box-shadow: 
-                inset 2px 2px 5px rgba(0, 0, 0, 0.3),
-                inset -2px -2px 5px rgba(255, 255, 255, 0.2);
-        }
         .stButton > button {
             width: 100%;
-            background: linear-gradient(
-                45deg,
-                #2563eb 0%,
-                #3b82f6 100%
-            );
+            background: linear-gradient(45deg, #2563eb 0%, #3b82f6 100%);
             color: white;
             border: none;
             border-radius: 16px;
@@ -102,45 +93,55 @@ def login():
             font-weight: bold;
             cursor: pointer;
             margin-top: 2rem;
-            box-shadow: 
-                0 8px 16px rgba(0, 0, 0, 0.4),
-                inset 0 -4px 8px rgba(0, 0, 0, 0.2),
-                inset 0 4px 8px rgba(255, 255, 255, 0.2);
+            box-shadow: 0 8px 16px rgba(0, 0, 0, 0.4),
+                        inset 0 -4px 8px rgba(0, 0, 0, 0.2),
+                        inset 0 4px 8px rgba(255, 255, 255, 0.2);
             transition: all 0.3s ease;
         }
         .stButton > button:hover {
             transform: translateY(-3px) scale(1.03);
-            box-shadow: 
-                0 12px 24px rgba(0, 0, 0, 0.5),
-                inset 0 -4px 8px rgba(0, 0, 0, 0.2),
-                inset 0 4px 8px rgba(255, 255, 255, 0.2);
-            background: linear-gradient(
-                45deg,
-                #1d4ed8 0%,
-                #2563eb 100%
-            );
+            box-shadow: 0 12px 24px rgba(0, 0, 0, 0.5),
+                        inset 0 -4px 8px rgba(0, 0, 0, 0.2),
+                        inset 0 4px 8px rgba(255, 255, 255, 0.2);
+            background: linear-gradient(45deg, #1d4ed8 0%, #2563eb 100%);
         }
         </style>
     """, unsafe_allow_html=True)
+
+    if 'password' not in st.session_state:
+        st.session_state.password = ""
     
     st.markdown("<h1>KAP NOTES</h1>", unsafe_allow_html=True)
-
+    
     client_names = get_client_names()
-
-    with st.form(key="login_form"):
-        client_name = st.selectbox("Select Client", client_names)
-        min_date = datetime(2000, 1, 1)
-        date_input = st.date_input("Enter the Date", min_value=min_date, max_value=datetime.today())
-        submit_button = st.form_submit_button("Sign In")
+    client_name = st.selectbox("Select Client", client_names)
+    
+    if client_name:
+        available_dates = get_dates_for_client(client_name)
+        selected_date = st.selectbox(f"Available Dates for {client_name}", available_dates)
         
-        if submit_button:
-            date_str = date_input.strftime("%d-%m-%Y")
-            date_str = date_str.replace("-", "_")
-            st.session_state.client_name = client_name
-            st.session_state.date = date_str
-            st.session_state.logged_in = True
-            st.rerun()
-
+        if selected_date:
+            available_meetings = get_meetings_for_date(client_name, selected_date)
+            selected_meeting = st.selectbox(f"Available Meetings for {selected_date}", available_meetings)
+            password = st.text_input("Enter Password", type="password", value=st.session_state.password)
+            sign_in_button = st.button("Sign In", key="sign_in")
+            if sign_in_button:
+                if password == "kapnotes12345":
+                    if validate_data(client_name, selected_date, selected_meeting):
+                        st.session_state.client_name = client_name
+                        st.session_state.date = selected_date
+                        st.session_state.meeting = selected_meeting
+                        st.session_state.logged_in = True
+                        st.session_state.password = password
+                        st.rerun()
+                    else:
+                        st.error(f"No records available for {client_name} on {selected_date}. Please select another option.")
+                elif not password:
+                    st.error("Please enter password.")
+                else: 
+                    st.error("Incorrect Password. Please try again.")
+                    st.session_state.password = password
+    
 
 if 'logged_in' not in st.session_state:
     st.session_state.logged_in = False
@@ -148,9 +149,14 @@ if 'logged_in' not in st.session_state:
 if not st.session_state.logged_in:
     login()
 else:
-    st.set_page_config(page_title="Kap Notes", layout="wide")
     client_name = st.session_state.client_name
     date = st.session_state.date
+    meeting = st.session_state.meeting
+    password= st.session_state.password
+ 
+    if st.sidebar.button("Back"):
+        st.session_state.logged_in = False
+        st.rerun() 
 
     st.sidebar.markdown(f'''
         <div class="client-name-container">
@@ -185,7 +191,7 @@ else:
             transition: transform 0.3s ease, box-shadow 0.3s ease;
         }
         .summary-box:hover, .keypoints-box:hover, .action-items-box:hover {
-            transform: translateY(-1px) scale(1.05);
+            transform: translateY(-0.1px) scale(1.05);
             box-shadow: 0px 12px 25px rgba(0, 0, 0, 0.25), 0px 18px 35px rgba(0, 0, 0, 0.2);
         }
         .summary-box {
@@ -357,9 +363,9 @@ else:
     '''
     st.markdown(css, unsafe_allow_html=True)
 
-    summary_blob_name = f"{client_name}/{date.replace('_', '-')}/summary.txt"
-    transcription_blob_name = f"{client_name}/{date.replace('_', '-')}/transcription.txt" 
-    audio_blob_name = f"{client_name}/{date.replace('_', '-')}/audio/audio.wav"
+    summary_blob_name = f"{client_name}/{date}/{meeting}/summary.txt"
+    transcription_blob_name = f"{client_name}/{date}/{meeting}/transcription.txt" 
+    audio_blob_name = f"{client_name}/{date}/{meeting}/audio.wav"
 
     bucket = client.bucket(bucket_name)
 
@@ -369,9 +375,17 @@ else:
     audio_blob = bucket.blob(audio_blob_name)
     audio_url = audio_blob.generate_signed_url(expiration=timedelta(hours=1), method='GET')
 
-    summary = re.search(r"Summary(.*?)\n\n", summary_content, re.DOTALL).group(1).strip()
-    key_points = re.findall(r"- (.*?)\n", re.search(r"Key Points(.*?)\n\n", summary_content, re.DOTALL).group(1))
-    action_items = re.findall(r"- (.*?)\n", re.search(r"Action Items(.*)", summary_content, re.DOTALL).group(1))
+    summary_match = re.search(r"Summary:\s*(.*?)(?=\nKey Points:)", summary_content, re.DOTALL)
+    summary = summary_match.group(1).strip() if summary_match else "Summary not found."
+
+    key_points_match = re.search(r"Key Points:\s*(.*?)(?=\nAction Items:)", summary_content, re.DOTALL)
+    key_points = re.findall(r"- (.*?)\n", key_points_match.group(1)) if key_points_match else ["Key points not found."]
+
+    action_items_match = re.search(r"Action Items:\s*(.*)", summary_content, re.DOTALL)
+    if action_items_match:
+        action_items = re.findall(r"- (.*?)(?=\n- |$)", action_items_match.group(1), re.DOTALL)
+    else:
+        action_items = ["Action items not found."]
 
     transcription_blob = bucket.blob(transcription_blob_name)
     with transcription_blob.open("r") as file:
@@ -524,5 +538,4 @@ else:
             xaxis=dict(title="Time (in seconds)"),
             yaxis=dict(title="Sentiment Score", range=[-1, 1]),
         )
-
         st.plotly_chart(fig)