initial commit

app.py

@@ -0,0 +1,53 @@
+import streamlit as st
+import requests
+
+st.title("Prompt Search Engine")
+st.write("Enter a prompt to find the top *n* similar prompts.")
+
+# User input prompt 
+prompt = st.text_input("Enter your prompt here:")
+
+# Input for n
+n = st.number_input("Number of similar prompts (n):", min_value=1, max_value=30, value=5, step=1)
+
+if st.button("Search"):
+    # Make sure the prompt is not empty
+    if prompt.strip() == "":
+        st.warning("Please enter a prompt.")
+    else:
+        # API endpoint
+        api_url = "https://anja97-prompt-search-engine.hf.space/search"
+
+        # Prepare the payload
+        payload = {
+            "query": prompt,
+            "n": int(n)
+        }
+
+        # Send the POST request
+        try:
+            with st.spinner('Searching for similar prompts...'):
+                response = requests.post(api_url, json=payload, timeout=30)
+            # Check if the request was successful
+            if response.status_code == 200:
+                data = response.json()
+                results = data.get("results", [])
+                if results:
+                    # Display the results
+                    st.success(f"Top {n} similar prompts:")
+                    for i, item in enumerate(results, start=1):
+                        score = item.get("score", 0)
+                        similar_prompt = item.get("prompt", "")
+                        st.markdown(f"### Result {i}")
+                        st.write(f"**Similarity score:** {score:.4f}")
+                        st.write(f"**Prompt:** {similar_prompt}")
+                        st.write("---")
+                else:
+                    st.info("No similar prompts found.")
+            else:
+                st.error(f"API Error: {response.status_code} - {response.text}")
+        except requests.exceptions.Timeout:
+            st.error("The request timed out. Please try again later.")
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
+

requirements.txt

@@ -0,0 +1,6 @@
+sentence-transformers
+numpy
+datasets
+fastapi
+uvicorn
+streamlit