Create app.py

app.py

@@ -0,0 +1,126 @@
+import os
+import gradio as gr
+import time
+
+from langchain.chat_models import ChatOpenAI
+
+# 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)
+        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)