add streamlit

Dockerfile

@@ -31,3 +31,4 @@ EXPOSE 7860
 
 # Command to run the FastAPI application
 CMD ["python", "run.py"]
+CMD ["streamlit", "run", "app.py", "--server.port 7860"]

app.py

@@ -0,0 +1,36 @@
+import streamlit as st
+import requests
+from typing import List, Tuple
+
+API_URL = "http://localhost:7860/most_similar"
+
+
+def get_similar_prompts(query: str, n: int) -> List[Tuple[float, str]]:
+    try:
+        response = requests.post(API_URL, json={"query": query, "n": n})
+        if response.status_code == 200:
+            return response.json().get("similar_prompts", [])
+        else:
+            st.error(f"Error: {response.status_code} - {response.text}")
+            return []
+    except requests.exceptions.RequestException as e:
+        st.error(f"Request error: {e}")
+        return []
+
+
+st.title("Prompt Search Engine")
+
+query = st.text_input("Enter a prompt:")
+n = st.slider("Number of similar prompts to retrieve:", 1, 20, 5)
+
+if st.button("Search"):
+    if query:
+        response = get_similar_prompts(query, n)
+        if response:
+            st.write(f"Top {n} similar prompts:")
+            for query_response in response:
+                st.write(
+                    f"**Score:** {query_response['score']:.4f} | **Prompt:** {query_response['prompt']}"
+                )
+    else:
+        st.warning("Please enter a prompt to search.")

requirements.txt

@@ -5,3 +5,4 @@ sentence-transformers==3.0.1
 numpy==1.26.4
 fastapi==0.111.1
 uvicorn==0.30.3
+streamlit==1.37.0