Update app.py

app.py

@@ -1,70 +1,40 @@
-import streamlit as st
-import dateparser
-from google.oauth2.credentials import Credentials
-from googleapiclient.discovery import build
+import gradio as gr
+from chatbot_schedule import schedule_bot_response
+from chatbot_wellness import wellness_bot_response
 
-# Google Calendar API setup
-SCOPES = ['https://www.googleapis.com/auth/calendar']
+# Define interfaces for each chatbot
+def task_schedule_interface(user_input):
+    return schedule_bot_response(user_input)
 
-def authenticate_google_calendar():
-    """Authenticate and return the Google Calendar service."""
-    creds = Credentials.from_authorized_user_file('token.json', SCOPES)
-    if not creds or not creds.valid:
-        st.error("Authentication required. Please set up your Google Calendar credentials.")
-        return None
-    return build('calendar', 'v3', credentials=creds)
+def wellness_chatbot_interface(user_id, user_input):
+    return wellness_bot_response(user_id, user_input)
 
-def add_event_to_calendar(service, title, start_datetime):
-    """Add an event to Google Calendar."""
-    event = {
-        'summary': title,
-        'start': {'dateTime': start_datetime, 'timeZone': 'UTC'},
-        'end': {'dateTime': start_datetime, 'timeZone': 'UTC'},
-    }
-    try:
-        event_result = service.events().insert(calendarId='primary', body=event).execute()
-        return event_result
-    except Exception as e:
-        st.error(f"Failed to add event: {e}")
-        return None
+# Create Gradio interfaces
+task_bot = gr.Interface(
+    fn=task_schedule_interface,
+    inputs=gr.Textbox(label="Describe your task and time (e.g., Schedule meeting for tomorrow at 6PM)"),
+    outputs="text",
+    title="Task Scheduler Chatbot",
+    description="Helps you schedule tasks into your Google Calendar."
+)
 
-def parse_input_and_schedule(input_text):
-    """Parse user input for title and datetime, then schedule the event."""
-    
-    # Split the input into words and extract the event title and date/time
-    words = input_text.split(' at ')  # Splitting around 'at' to separate title and time part
-    if len(words) < 2:
-        st.error("Could not parse the event. Make sure to specify the title and time (e.g., 'gym for tomorrow at 6pm' or 'meeting for December 7 at 8:30am').")
-        return
-    
-    title = words[0].strip()  # The event title
-    date_input = ' at '.join(words[1:]).strip()  # The time portion, including relative or absolute date
-    
-    # Parse the date/time using dateparser
-    parsed_date = dateparser.parse(date_input, settings={'PREFER_DATES_FROM': 'future'})
-    
-    # If the date is parsed as the next day (incorrect), we can force it to use the exact date
-    if not parsed_date:
-        st.error(f"Could not parse the date and time: {date_input}. Please try again.")
-        return
+wellness_bot = gr.Interface(
+    fn=wellness_chatbot_interface,
+    inputs=[
+        gr.Textbox(label="Enter your unique user ID (e.g., email, username)"),
+        gr.Textbox(label="What would you like to discuss?"),
+    ],
+    outputs="text",
+    title="Mental Wellness Chatbot",
+    description="Supports your mental wellness with tailored motivational advice."
+)
 
-    # If parsed_date is earlier than today, it may be misinterpreted, so we adjust
-    today = dateparser.parse("today")
-    if parsed_date.date() < today.date():
-        st.error(f"The date seems to be incorrect. Please check the input format for specific dates like 'December 7 at 3:34pm'.")
-        return
-    
-    # Show the parsed event details
-    st.write(f"Event Title: {title}")
-    st.write(f"Scheduled Time: {parsed_date}")
-    
-    # Add event to Google Calendar (this would be the integration part)
-    st.success(f"Event '{title}' scheduled for {parsed_date}!")
+# Combine into a tabbed interface
+app = gr.TabbedInterface(
+    interfaces=[task_bot, wellness_bot],
+    tab_names=["Task Scheduler", "Mental Wellness Support"]
+)
 
-# Streamlit App
-st.title("Smart Scheduler with Google Calendar")
-st.write("Add tasks to your schedule using natural language.")
-
-user_input = st.text_input("What would you like to schedule?", "")
-if user_input:
-    parse_input_and_schedule(user_input)
+# Launch the app
+if __name__ == "__main__":
+    app.launch()