Vitomir Jovanović commited on
Commit
0f0853e
·
1 Parent(s): 06940e7

Update requirements

Browse files
Files changed (3) hide show
  1. api.py +0 -33
  2. environment.yaml +0 -0
  3. requirements.txt +0 -0
api.py DELETED
@@ -1,33 +0,0 @@
1
- # streamlit_app.py
2
- import streamlit as st
3
- import requests
4
-
5
- # Streamlit app title
6
- st.title("Top K Search with Vector DataBase")
7
-
8
- # FastAPI endpoint URL
9
- url = "http://127.0.0.1:8000/search/"
10
-
11
- # Input fields in Streamlit
12
- id = st.text_input("Enter ID:", value="1")
13
- prompt = st.text_input("Enter your prompt:")
14
- k = st.number_input("Top K results:", min_value=1, max_value=100, value=3)
15
-
16
- # Trigger the search when the button is clicked
17
- if st.button("Search"):
18
- # Construct the request payload
19
- payload = {
20
- "id": id,
21
- "prompt": prompt,
22
- "k": k
23
- }
24
-
25
- # Make the POST request
26
- response = requests.post(url, json=payload)
27
-
28
- # Handle the response
29
- if response.status_code == 200:
30
- results = response.json()
31
- st.write(results)
32
- else:
33
- st.error(f"Error: {response.status_code} - {response.text}")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
environment.yaml CHANGED
Binary files a/environment.yaml and b/environment.yaml differ
 
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ