Update app.py

app.py

@@ -1,5 +1,21 @@
-import streamlit as st
+import os
 import requests
+from PIL import Image, ImageDraw, ImageFont
+from io import BytesIO
+import base64
+from utils import image_interface
+import streamlit as st
+from streamlit_image_select import image_select
+
+AUTH_TOKEN = st. secrets["AUTH_TOKEN"]
+
+headers = {"Authorization": f"Bearer {AUTH_TOKEN}"}
+st.set_page_config(page_title="Image Similarity", page_icon="🎆🎇")
+CHAT_API = "https://abhilashvj-computer-vision-backend.hf.space/"
+# IMAGE_SIMILARITY_DEMO = "http://127.0.0.1:8000/find-similar-image-pinecone/"
+TMP_DIR = "./tmp"
+
+os.makedirs(TMP_DIR, exist_ok=True)
 
 st.title("Document Uploader and Chatbot Trainer")
 
@@ -9,7 +25,7 @@ st.header("Upload Documents")
 evergreen_file = st.file_uploader("Choose an Evergreen Document", type=['txt', 'pdf', 'doc', 'docx'])
 if evergreen_file:
     files = {'file': evergreen_file.getvalue()}
-    response = requests.post('http://your_fastapi_endpoint/upload/evergreen/', files=files)
+    response = requests.post(f'{CHAT_API}upload/evergreen/', files=files, headers=headers)
     if response.json().get("success"):
         st.success("Evergreen document uploaded successfully!")
     else:
@@ -19,25 +35,25 @@ if evergreen_file:
 dynamic_file = st.file_uploader("Choose a Dynamic Document", type=['txt', 'pdf', 'doc', 'docx'])
 if dynamic_file:
     files = {'file': dynamic_file.getvalue()}
-    response = requests.post('http://your_fastapi_endpoint/upload/dynamic/', files=files)
+    response = requests.post(f'{CHAT_API}upload/dynamic/', files=files, headers=headers)
     if response.json().get("success"):
         st.success("Dynamic document uploaded successfully!")
     else:
         st.error("Failed to upload dynamic document!")
 
 # Train bot button
-if st.button("Train Bot"):
-    response = requests.post('http://your_fastapi_endpoint/train/')
-    bot_url = response.json().get("bot_url")
-    if bot_url:
-        st.success(f"Bot trained successfully! Access the bot at {bot_url}")
-    else:
-        st.error("Failed to train the bot!")
+# if st.button("Train Bot"):
+#     response = requests.post('http://your_fastapi_endpoint/train/')
+#     bot_url = response.json().get("bot_url")
+#     if bot_url:
+#         st.success(f"Bot trained successfully! Access the bot at {bot_url}")
+#     else:
+#         st.error("Failed to train the bot!")
 
 # Chat with bot
 st.header("Chat with Bot")
 user_input = st.text_input("Ask your question:")
 # Assuming you have an endpoint to send user questions and get responses
-# response = requests.post('http://your_fastapi_endpoint/ask/', json={"question": user_input})
-# bot_response = response.json().get("answer")
-# st.text_area("Bot's Response:", value=bot_response)
+response = requests.post(f'{CHAT_API}/ask/', json={"question": user_input})
+bot_response = response.json().get("answer")
+st.text_area("Bot's Response:", value=bot_response)