### straeamlit_app.py ###
import os
import streamlit as st
import requests

### Streamlit App ###
### configuration ###
st.set_page_config(
    page_title="Real Estate Chatbot",
    page_icon="🏠",
    layout="centered",
    initial_sidebar_state="auto",
)

st.title("Real Estate Chatbot")

# Use environment variable for DEBUG mode
DEBUG = os.getenv("DJANGO_DEBUG", "True") == "True"

if DEBUG:
    url = "http://localhost:8000/rag/chatbot/"
else:
    url = "https://api.remax.rdweb.com.br/rag/chatbot/"

user_input = st.text_input("Ask about properties:")

if user_input:
    response = requests.post(
        url,
        json={"prompt": user_input}
    )
    
    # Check if the request was successful
    if response.status_code == 200:
        try:
            response_data = response.json()
            st.write(response_data.get('response', 'No response key found in JSON'))
        except ValueError:
            st.write("Failed to decode JSON response.")
    else:
        st.write(f"Request failed with status code {response.status_code}")