Update app.py

app.py

@@ -261,48 +261,88 @@ def fetch_upcoming_events(creds, start_date=None, end_date=None):
         return []
 
 # --- GMAIL OPERATIONS ---
-def fetch_unread_emails(service):
+# def fetch_unread_emails(service):
+#     structured_emails = []
+#     try:
+#         results = service.users().messages().list(userId="me", maxResults=5, q="is:unread").execute()
+#         for m_info in results.get("messages", []):
+#             msg = service.users().messages().get(userId='me', id=m_info['id']).execute()
+#             headers = {h['name']: h['value'] for h in msg['payload']['headers']}
+
+#             body = ""
+#             payload = msg.get('payload', {})
+
+#             def get_body(parts):
+#                 for part in parts:
+#                     mimeType = part.get('mimeType')
+#                     data = part.get('body', {}).get('data')
+#                     if mimeType == 'text/plain' and data:
+#                         return base64.urlsafe_b64decode(data).decode('utf-8')
+#                     elif 'parts' in part:
+#                         result = get_body(part['parts'])
+#                         if result: return result
+#                 return ""
+
+#             if 'parts' in payload:
+#                 body = get_body(payload['parts'])
+#             else:
+#                 data = payload.get('body', {}).get('data')
+#                 if data:
+#                     body = base64.urlsafe_b64decode(data).decode('utf-8')
+
+#             structured_emails.append({
+#                 'id': msg['id'],
+#                 'time': time.strftime('%Y-%m-%d %I:%M %p', time.localtime(int(msg['internalDate'])/1000)),
+#                 'from': headers.get('From', 'N/A'),
+#                 'subject': headers.get('Subject', 'N/A'),
+#                 'body': body if body else "No readable text found",
+#                 'summary': ''
+#             })
+#         return structured_emails
+#     except Exception as e:
+#         st.error(f"Error fetching emails: {e}")
+#         return []
+
+
+
+def fetch_emails_by_date(service, target_date, max_results=50):
     structured_emails = []
+    # Format dates for Gmail query (YYYY/MM/DD)
+    after_date = target_date.strftime('%Y/%m/%d')
+    before_date = (target_date + timedelta(days=1)).strftime('%Y/%m/%d')
+    
+    # Query: Emails received on the specific day
+    query = f"after:{after_date} before:{before_date}"
+    
     try:
-        results = service.users().messages().list(userId="me", maxResults=5, q="is:unread").execute()
-        for m_info in results.get("messages", []):
+        results = service.users().messages().list(userId="me", maxResults=max_results, q=query).execute()
+        messages = results.get("messages", [])
+        
+        if not messages:
+            return []
+
+        for m_info in messages:
             msg = service.users().messages().get(userId='me', id=m_info['id']).execute()
             headers = {h['name']: h['value'] for h in msg['payload']['headers']}
-
-            body = ""
-            payload = msg.get('payload', {})
-
-            def get_body(parts):
-                for part in parts:
-                    mimeType = part.get('mimeType')
-                    data = part.get('body', {}).get('data')
-                    if mimeType == 'text/plain' and data:
-                        return base64.urlsafe_b64decode(data).decode('utf-8')
-                    elif 'parts' in part:
-                        result = get_body(part['parts'])
-                        if result: return result
-                return ""
-
-            if 'parts' in payload:
-                body = get_body(payload['parts'])
-            else:
-                data = payload.get('body', {}).get('data')
-                if data:
-                    body = base64.urlsafe_b64decode(data).decode('utf-8')
+            
+            # Simplified body extraction logic (re-use your existing get_body logic here)
+            # ... [Body extraction code] ...
 
             structured_emails.append({
                 'id': msg['id'],
-                'time': time.strftime('%Y-%m-%d %I:%M %p', time.localtime(int(msg['internalDate'])/1000)),
+                'time': datetime.datetime.fromtimestamp(int(msg['internalDate'])/1000).strftime('%I:%M %p'),
                 'from': headers.get('From', 'N/A'),
                 'subject': headers.get('Subject', 'N/A'),
-                'body': body if body else "No readable text found",
+                'body': "Extracted Body Text...", # Replace with actual extraction
                 'summary': ''
             })
         return structured_emails
     except Exception as e:
-        st.error(f"Error fetching emails: {e}")
+        st.error(f"Error fetching emails for {after_date}: {e}")
         return []
 
+        
+
 # --- AI AGENT PARSERS ---
 def parse_schedule_prompt(prompt, current_date, user_timezone_str):
     tz = pytz.timezone(user_timezone_str)
@@ -475,19 +515,85 @@ def main():
     # ---------------------------------------------------------
     # TAB 1: EMAIL INBOX ASSISTANT
     # ---------------------------------------------------------
+    # with tab1:
+    #     st.header("1. 📧 Agent 1: Email Fetching")
+        
+    #     # --- Step 1: Fetch Raw Data ---
+    #     if st.button("▶️ Step 1: Fetch Raw Email Data (Gmail)"):
+    #         gmail_service = build("gmail", "v1", credentials=creds)
+    #         with st.spinner("Connecting to Gmail and fetching structured email data..."):
+    #             st.session_state.email_data = fetch_unread_emails(gmail_service) 
+    #             st.session_state.summarized = False 
+        
+    #     # --- Display Agent 1 Results ---
+    #     if st.session_state.email_data:
+    #         st.subheader("Raw Data Extracted from Gmail")
+    #         raw_table_data = [
+    #             {
+    #                 'Date & Time': e['time'], 
+    #                 'From': e['from'], 
+    #                 'Subject': e['subject'], 
+    #                 'Body': e['body'],
+    #                 'Attachments': 'Yes' if e.get('attachments') else 'No'
+    #             }
+    #             for e in st.session_state.email_data
+    #         ]
+            
+    #         st.dataframe(
+    #             raw_table_data, 
+    #             use_container_width=True,
+    #             column_order=('Date & Time', 'From', 'Subject', 'Body', 'Attachments'), 
+    #             column_config={
+    #                 "Subject": st.column_config.Column(width="large"),
+    #                 "Body": st.column_config.TextColumn(
+    #                     "Email Body (Scroll/Click)", 
+    #                     help="The full text of the email. Click to read all content.",
+    #                     width="medium",
+    #                     max_chars=200, 
+    #                 )
+    #             }
+    #         )
+            
+    #         st.divider()
+
     with tab1:
         st.header("1. 📧 Agent 1: Email Fetching")
         
-        # --- Step 1: Fetch Raw Data ---
-        if st.button("▶️ Step 1: Fetch Raw Email Data (Gmail)"):
+        # --- NEW: Date Navigation UI (Similar to your Calendar navigation) ---
+        col_prev, col_date, col_next = st.columns([1, 2, 1])
+        
+        if col_prev.button("⬅️ Previous Day", use_container_width=True):
+            st.session_state.email_selected_date -= timedelta(days=1)
+            st.rerun()
+            
+        # Allow manual date selection too
+        st.session_state.email_selected_date = col_date.date_input(
+            "Target Date", 
+            value=st.session_state.email_selected_date,
+            label_visibility="collapsed"
+        )
+            
+        if col_next.button("Next Day ➡️", use_container_width=True):
+            st.session_state.email_selected_date += timedelta(days=1)
+            st.rerun()
+
+        st.info(f"📅 Currently targeting emails for: **{st.session_state.email_selected_date.strftime('%Y-%m-%d')}**")
+        
+        # --- Modified Step 1: Use the new fetch_emails_by_date ---
+        if st.button(f"▶️ Step 1: Fetch Emails for {st.session_state.email_selected_date}", use_container_width=True):
             gmail_service = build("gmail", "v1", credentials=creds)
-            with st.spinner("Connecting to Gmail and fetching structured email data..."):
-                st.session_state.email_data = fetch_unread_emails(gmail_service) 
+            with st.spinner(f"Searching Gmail for {st.session_state.email_selected_date}..."):
+                # Calling your new function instead of fetch_unread_emails
+                st.session_state.email_data = fetch_emails_by_date(
+                    gmail_service, 
+                    st.session_state.email_selected_date,
+                    max_results=50 # Increased for high-volume handling
+                ) 
                 st.session_state.summarized = False 
         
-        # --- Display Agent 1 Results ---
+        # --- Display Agent 1 Results (No changes to your view logic here) ---
         if st.session_state.email_data:
-            st.subheader("Raw Data Extracted from Gmail")
+            st.subheader(f"Raw Data Extracted ({len(st.session_state.email_data)} emails)")
             raw_table_data = [
                 {
                     'Date & Time': e['time'], 
@@ -507,7 +613,6 @@ def main():
                     "Subject": st.column_config.Column(width="large"),
                     "Body": st.column_config.TextColumn(
                         "Email Body (Scroll/Click)", 
-                        help="The full text of the email. Click to read all content.",
                         width="medium",
                         max_chars=200, 
                     )