import os import streamlit as st import requests import msal from datetime import datetime, timedelta import calendar # Configuration APPLICATION_ID_KEY = os.getenv('APPLICATION_ID_KEY') CLIENT_SECRET_KEY = os.getenv('CLIENT_SECRET_KEY') AUTHORITY_URL = 'https://login.microsoftonline.com/common' REDIRECT_URI = 'https://huggingface.co/spaces/awacke1/MSGraphAPI' # Define product to scope mapping and links PRODUCT_SCOPES = { "📧 Outlook": {'scopes': ['Mail.Read', 'Mail.Send'], 'link': 'https://outlook.office.com/mail/'}, "📅 Calendar": {'scopes': ['Calendars.ReadWrite'], 'link': 'https://outlook.office.com/calendar/'}, "📋 Tasks": {'scopes': ['Tasks.ReadWrite'], 'link': 'https://to-do.office.com/tasks/'}, "🗂️ OneDrive": {'scopes': ['Files.ReadWrite.All'], 'link': 'https://onedrive.live.com/'}, # ... (other products) } BASE_SCOPES = ['User.Read'] # ... (keep the get_msal_app and get_access_token functions as they were) ... def make_api_call(access_token, endpoint, method='GET', data=None): headers = {'Authorization': f'Bearer {access_token}', 'Content-Type': 'application/json'} url = f'https://graph.microsoft.com/v1.0/{endpoint}' if method == 'GET': response = requests.get(url, headers=headers) elif method == 'POST': response = requests.post(url, headers=headers, json=data) else: raise ValueError(f"Unsupported method: {method}") if response.status_code in [200, 201]: return response.json() else: st.error(f"API call failed: {response.status_code} - {response.text}") return None def handle_outlook_integration(access_token): st.subheader("📧 Outlook Integration") st.markdown(f"[Open Outlook]({PRODUCT_SCOPES['📧 Outlook']['link']})") emails = make_api_call(access_token, 'me/messages?$top=10&$orderby=receivedDateTime desc') if emails and 'value' in emails: for email in emails['value']: with st.expander(f"From: {email['from']['emailAddress']['name']} - Subject: {email['subject']}"): st.write(f"Received: {email['receivedDateTime']}") st.write(f"Body: {email['bodyPreview']}") else: st.write("No emails found or unable to fetch emails.") def handle_calendar_integration(access_token): st.subheader("📅 Calendar Integration") st.markdown(f"[Open Calendar]({PRODUCT_SCOPES['📅 Calendar']['link']})") # Get the current month's start and end dates now = datetime.now() start_of_month = now.replace(day=1, hour=0, minute=0, second=0, microsecond=0) end_of_month = start_of_month.replace(month=start_of_month.month % 12 + 1, day=1) - timedelta(days=1) events = make_api_call(access_token, f"me/calendarView?startDateTime={start_of_month.isoformat()}&endDateTime={end_of_month.isoformat()}&$orderby=start/dateTime") if events and 'value' in events: # Create a calendar view cal = calendar.monthcalendar(now.year, now.month) st.write(f"Calendar for {now.strftime('%B %Y')}") # Create a placeholder for each day day_placeholders = {} for week in cal: cols = st.columns(7) for i, day in enumerate(week): if day != 0: day_placeholders[day] = cols[i].empty() day_placeholders[day].write(f"**{day}**") # Populate the calendar with events for event in events['value']: start_date = datetime.fromisoformat(event['start']['dateTime'][:-1]) # Remove 'Z' from the end day = start_date.day if day in day_placeholders: day_placeholders[day].write(f"{start_date.strftime('%H:%M')} - {event['subject']}") else: st.write("No events found or unable to fetch events.") st.write("Add a new event:") event_subject = st.text_input("Event Subject") event_date = st.date_input("Event Date") event_time = st.time_input("Event Time") if st.button("Add Event"): event_start = datetime.combine(event_date, event_time) event_end = event_start + timedelta(hours=1) new_event = { "subject": event_subject, "start": { "dateTime": event_start.isoformat(), "timeZone": "UTC" }, "end": { "dateTime": event_end.isoformat(), "timeZone": "UTC" } } result = make_api_call(access_token, 'me/events', method='POST', data=new_event) if result: st.success("Event added successfully!") else: st.error("Failed to add event.") def handle_tasks_integration(access_token): st.subheader("📋 Tasks Integration") st.markdown(f"[Open Tasks]({PRODUCT_SCOPES['📋 Tasks']['link']})") tasks = make_api_call(access_token, 'me/todo/lists') if tasks and 'value' in tasks: default_list = next((list for list in tasks['value'] if list['wellknownListName'] == 'defaultList'), None) if default_list: tasks = make_api_call(access_token, f"me/todo/lists/{default_list['id']}/tasks") if tasks and 'value' in tasks: for task in tasks['value']: st.write(f"Task: {task['title']}") st.write(f"Status: {'Completed' if task['status'] == 'completed' else 'Not Completed'}") st.write("---") else: st.write("No tasks found or unable to fetch tasks.") else: st.write("Default task list not found.") else: st.write("Unable to fetch task lists.") st.write("Add a new task:") task_title = st.text_input("Task Title") if st.button("Add Task"): new_task = { "title": task_title } result = make_api_call(access_token, f"me/todo/lists/{default_list['id']}/tasks", method='POST', data=new_task) if result: st.success("Task added successfully!") else: st.error("Failed to add task.") def handle_onedrive_integration(access_token): st.subheader("🗂️ OneDrive Integration") st.markdown(f"[Open OneDrive]({PRODUCT_SCOPES['🗂️ OneDrive']['link']})") files = make_api_call(access_token, 'me/drive/root/children') if files and 'value' in files: for file in files['value']: st.write(f"Name: {file['name']}") st.write(f"Type: {'Folder' if 'folder' in file else 'File'}") st.write(f"Last Modified: {file['lastModifiedDateTime']}") st.write("---") else: st.write("No files found or unable to fetch files.") def main(): st.title("🦄 MS Graph API with AI & Cloud Integration for M365") st.sidebar.title("📝 M365 Products") st.sidebar.write("Select products to integrate:") selected_products = {} for product in PRODUCT_SCOPES.keys(): selected = st.sidebar.checkbox(product) if selected: selected_products[product] = True request_scopes = BASE_SCOPES.copy() for product in selected_products: request_scopes.extend(PRODUCT_SCOPES[product]['scopes']) request_scopes = list(set(request_scopes)) st.session_state['request_scopes'] = request_scopes if 'access_token' not in st.session_state: client_instance = get_msal_app() auth_url = client_instance.get_authorization_request_url( scopes=request_scopes, redirect_uri=REDIRECT_URI ) st.write('👋 Please [click here]({}) to log in and authorize the app.'.format(auth_url)) query_params = st.query_params if 'code' in query_params: code = query_params.get('code') st.write('🔑 Authorization Code Obtained:', code[:10] + '...') try: access_token = get_access_token(code) st.session_state['access_token'] = access_token st.success("Access token acquired successfully!") st.rerun() except Exception as e: st.error(f"Error acquiring access token: {str(e)}") st.stop() else: access_token = st.session_state['access_token'] user_info = make_api_call(access_token, 'me') if user_info: st.sidebar.write(f"👋 Hello, {user_info.get('displayName', 'User')}!") if selected_products: for product in selected_products: if product == "📧 Outlook": handle_outlook_integration(access_token) elif product == "📅 Calendar": handle_calendar_integration(access_token) elif product == "📋 Tasks": handle_tasks_integration(access_token) elif product == "🗂️ OneDrive": handle_onedrive_integration(access_token) # Add more product integrations here else: st.write("No products selected. Please select products from the sidebar.") if __name__ == "__main__": main()