Update app.py

app.py

@@ -1,126 +1,124 @@
 import os
+import re
+from datetime import datetime, timedelta
+import openai
 import gradio as gr
-import time
-
-from langchain.chat_models import ChatOpenAI
+from ics import Calendar, Event
 
 # Set your OpenAI API key
-os.environ["OPENAI_API_KEY"] = "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"
-
-# Initialize the ChatOpenAI model
-chat_model = ChatOpenAI(model="gpt-3.5-turbo-1106", temperature=0.2)
-
-# Define the apartment details dictionary
-apartment_details = {
-    "1": {
-        "name": "Luxury Apartment in Downtown",
-        "address": "123 Main St, Anytown, USA",
-        "bedrooms": 2,
-        "bathrooms": 2,
-        "sqft": 1200,
-        "price": "$2000/month",
-        "description": "Beautiful luxury apartment in the heart of downtown. Spacious living area, modern kitchen, and stunning views."
-    },
-    "2": {
-        "name": "Cozy Studio near Park",
-        "address": "456 Elm St, Anytown, USA",
-        "bedrooms": 0,
-        "bathrooms": 1,
-        "sqft": 500,
-        "price": "$1000/month",
-        "description": "Charming studio apartment located near the park. Perfect for a single person or couple. Close to amenities."
-    },
-    "3": {
-        "name": "Family-Friendly Townhouse",
-        "address": "789 Oak St, Anytown, USA",
-        "bedrooms": 3,
-        "bathrooms": 2.5,
-        "sqft": 1800,
-        "price": "$2500/month",
-        "description": "Spacious townhouse with plenty of room for a family. Close to schools, parks, and shopping centers."
-    }
-}
-
-def provide_information(apartment_id=None):
-    if apartment_id:
-        apartment_info = apartment_details.get(apartment_id)
-        if apartment_info:
-            return (
-                f"Name: {apartment_info['name']}\n"
-                f"Address: {apartment_info['address']}\n"
-                f"Bedrooms: {apartment_info['bedrooms']}\n"
-                f"Bathrooms: {apartment_info['bathrooms']}\n"
-                f"Square Footage: {apartment_info['sqft']} sqft\n"
-                f"Price: {apartment_info['price']}\n"
-                f"Description: {apartment_info['description']}"
-            )
-        else:
-            return "Apartment not found"
-    else:
-        all_apartments_info = ""
-        for apartment_id, apartment_info in apartment_details.items():
-            all_apartments_info += (
-                f"Apartment ID: {apartment_id}\n"
-                f"Name: {apartment_info['name']}\n"
-                f"Address: {apartment_info['address']}\n"
-                f"Bedrooms: {apartment_info['bedrooms']}\n"
-                f"Bathrooms: {apartment_info['bathrooms']}\n"
-                f"Square Footage: {apartment_info['sqft']} sqft\n"
-                f"Price: {apartment_info['price']}\n"
-                f"Description: {apartment_info['description']}\n\n"
-            )
-        return all_apartments_info
-
-def make_purchase_reservation(apartment_id: str, documents: dict):
-    # Print the provided data and a success message
-    print(f"Apartment ID: {apartment_id}, Documents: {documents}")
-    return "SUCCESS"
-
-def execute_purchase(apartment_id, documents):
-    # For simplicity, let's just print the data provided and a success message
-    print(f"Apartment ID: {apartment_id}, Documents: {documents}")
-    return make_purchase_reservation(apartment_id, documents)
-
-def update_owner_information(owner_id, new_number):
-    # For simplicity, let's just print the data provided and a success message
-    print(f"Owner ID: {owner_id}, New Number: {new_number}")
-    return "Owner information updated successfully."
-
-def handle_message(message, history=None):
-    # Split message into parts
-    parts = message.split()
-    
-    # Check for different categories of problems
-    if "provide information" in message:
-        if "on Apartment" in message:
-            apartment_id = parts[-1]
-            return provide_information(apartment_id)
+openai.api_key = ("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")
+
+# Define a custom calendar class
+class CustomCalendar:
+    def __init__(self, file_path):
+        self.file_path = file_path
+        self.calendar = Calendar()
+
+    def create_event(self, title, start_time, end_time):
+        event = Event()
+        event.name = title
+        event.begin = start_time
+        event.end = end_time
+        self.calendar.events.add(event)
+        self.save_to_file()
+
+    def update_event(self, old_title, new_start_time, new_end_time):
+        for event in self.calendar.events:
+            if event.name == old_title:
+                event.begin = new_start_time
+                event.end = new_end_time
+                self.save_to_file()
+                break
+
+    def get_events(self):
+        return list(self.calendar.events)
+
+    def save_to_file(self):
+        with open(self.file_path, "w") as file:
+            file.writelines(self.calendar)
+
+# Helper function to extract event details from the chatbot's response
+def extract_event_details(response):
+    time_pattern = r"(\d{1,2})\s*([ap]m)"
+    title_pattern = r"(for|about)\s+(.+)"
+
+    time_match = re.search(time_pattern, response, re.IGNORECASE)
+    title_match = re.search(title_pattern, response, re.IGNORECASE)
+
+    if time_match:
+        hour = int(time_match.group(1))
+        meridiem = time_match.group(2).lower()
+        if meridiem == "pm" and hour < 12:
+            hour += 12
+        elif meridiem == "am" and hour == 12:
+            hour = 0
+
+    if title_match:
+        title = title_match.group(2)
+
+    if time_match and title_match:
+        start_time = datetime.now().replace(hour=hour, minute=0, second=0, microsecond=0)
+        end_time = start_time + timedelta(hours=1)
+        return title, start_time, end_time
+
+    return None, None, None
+
+# Function to handle user messages
+def handle_message(texts):
+    global calendar
+    responses = []
+
+    for text in texts:
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": text}],
+            temperature=0.7,
+        )
+        response_text = response.choices[0].message.content
+
+        # Check for event creation or adjustment
+        title, start_time, end_time = extract_event_details(response_text)
+        if title and start_time and end_time:
+            if "create" in text or "set" in text or "make" in text:
+                calendar.create_event(title, start_time, end_time)
+                responses.append(f"Appointment created: {title} at {start_time.strftime('%I:%M %p')}")
+            elif "change" in text or "adjust" in text or "modify" in text:
+                old_title = None
+                for event in calendar.get_events():
+                    if event.name.lower() in response_text.lower():
+                        old_title = event.name
+                        break
+                if old_title:
+                    calendar.update_event(old_title, start_time, end_time)
+                    responses.append(f"Appointment updated: {old_title} to {start_time.strftime('%I:%M %p')}")
+                else:
+                    responses.append("Sorry, I couldn't find the appointment you wanted to adjust.")
+        elif "show" in response_text or "list" in response_text or "view" in response_text:
+            events = calendar.get_events()
+            if events:
+                event_list = "\n".join([f"{event.name}: {event.begin.strftime('%I:%M %p')} - {event.end.strftime('%I:%M %p')}" for event in events])
+                responses.append(f"Your appointments:\n{event_list}")
+            else:
+                responses.append("You don't have any appointments scheduled.")
         else:
-            return provide_information()
-    elif "execute purchase" in message:
-        apartment_id, *documents = parts[-2:]
-        return execute_purchase(apartment_id, " ".join(documents))
-    elif "update owner information" in message:
-        owner_id, new_number = parts[-2:]
-        return update_owner_information(owner_id, new_number)
-    elif "list all houses" in message:
-        return provide_information()
-    else:
-        # Use the LangChain model for general chat
-        response = chat_model.predict(message)
-        return response
-
-# Create a Gradio chat interface
-gr.ChatInterface(
-    handle_message,
-    chatbot=gr.Chatbot(height=300),
-    textbox=gr.Textbox(placeholder="How can I help you today?", container=True, scale=8),
-    title="Joe's Real Estate Chatbot",
-    description="Ask D'Law any law related question",
-    theme="compact",
-    examples=["list all houses", "provide information on Apartment 2", "execute purchase on apartment 2", "update owner information on apartment"],
-    retry_btn=None,
-    submit_btn='Send',
-    undo_btn=False,
-    clear_btn="Clear",
-).launch(share=True)
+            responses.append(response_text)
+
+    return responses
+
+# Define calendar file path
+calendar_file_path = "calendar.ics"
+
+# Create a custom calendar instance
+calendar = CustomCalendar(calendar_file_path)
+
+# Define Gradio interface
+iface = gr.Interface(
+    fn=handle_message,
+    inputs=gr.Textbox(lines=3, label="Enter your messages (one per line)"),
+    outputs="text",
+    title="Calendar Chatbot",
+    description="Interact with the chatbot to manage your calendar events.",
+)
+
+# Run the Gradio interface
+iface.launch(debug=True)