construct url on the fly

app.py

@@ -1,9 +1,19 @@
+import os
+
 import requests
 import streamlit as st
+from dotenv import load_dotenv
 
 from responses import SubmitQuestionAndDocumentsResponse
 
 st.set_page_config(layout="wide")
+load_dotenv()
+
+if os.getenv("ENV") == "production":
+    BASE_URL = "https://deven-cleric-backend.onrender.com"
+else:
+    BASE_URL = "http://localhost:8000"
+
 
 def make_sidebar():
     with st.sidebar:
@@ -11,23 +21,23 @@ def make_sidebar():
         st.write("This is a sidebar.")
         st.write("You can add widgets here")
 
+
 def create_payload(question, logs):
-    payload = {
-        "question": question,
-        "logs": logs
-    }
+    payload = {"question": question, "logs": logs}
     # st.write(payload)
     return payload
 
+
 def process_payload(payload):
     all_logs = payload["logs"]
     text = ""
     for log in all_logs:
         text += requests.get(log).text
 
-    payload['text'] = text
+    payload["text"] = text
     return payload
 
+
 def main():
     st.title("Hello, World!")
     # print("Hello, World!")
@@ -37,15 +47,20 @@ def main():
 
     with col1:
         st.write("This is column 2")
-        logs = st.multiselect("Select the options",
-                              [
-                                  "https://storage.googleapis.com/cleric-assignment-call-logs/call_log_20240314_104111.txt",
-                                  "https://storage.googleapis.com/cleric-assignment-call-logs/call_log_20240315_104111.txt",
-                                  "https://storage.googleapis.com/cleric-assignment-call-logs/call_log_20240316_104111.txt",])
+        logs = st.multiselect(
+            "Select the options",
+            [
+                "https://storage.googleapis.com/cleric-assignment-call-logs/call_log_20240314_104111.txt",
+                "https://storage.googleapis.com/cleric-assignment-call-logs/call_log_20240315_104111.txt",
+                "https://storage.googleapis.com/cleric-assignment-call-logs/call_log_20240316_104111.txt",
+            ],
+        )
 
     with col2:
         st.write("This is column 1")
-        question = st.text_input("Ask the question", value="What product design decisions did the team make?")
+        question = st.text_input(
+            "Ask the question", value="What product design decisions did the team make?"
+        )
 
     payload = create_payload(question, logs)
     processed_payload = process_payload(payload)
@@ -55,11 +70,17 @@ def main():
     st.write(data.model_dump())
 
     if st.button("Submit"):
-        url = "https://deven-cleric-backend.onrender.com/submit_question_and_documents/"
-        url_local = "http://"
+        # url = "https://deven-cleric-backend.onrender.com/submit_question_and_documents/"
+        url = f"{BASE_URL}/submit_question_and_documents/"
         resp = requests.post(url, json=data.model_dump())
         st.write(resp.status_code)
         st.write(resp.json())
 
+        url_local_get = f"{BASE_URL}/get_question_and_facts/"
+        resp = requests.get(url_local_get)
+        st.write(resp.status_code)
+        st.write(resp.json())
+
+
 if __name__ == "__main__":
-    main()
+    main()