Spaces:

lazarr19
/

prompt-engine

Sleeping

App Files Files Community

Lazar Radojevic commited on Jul 28, 2024

Commit

45462fb

1 Parent(s): be043a6

finally working version :)

Browse files

Files changed (3) hide show

frontend/app_ui.py +18 -4
main.py +34 -0
poe/common-tasks.toml +11 -2

frontend/app_ui.py CHANGED Viewed

@@ -2,7 +2,7 @@ import streamlit as st
 import requests
 # Define the URL of the FastAPI service
-API_URL = "http://localhost:8000/most_similar"  # Adjust if the FastAPI service is hosted elsewhere
 def get_similar_prompts(query, n):
@@ -15,13 +15,22 @@ def get_similar_prompts(query, n):
         return None
 def main():
     st.title("Prompt Similarity Finder")
     # User input for query
     query = st.text_input("Enter your query:", "")
     n = st.slider(
-        "Number of similar prompts to retrieve:", min_value=1, max_value=20, value=5
     )
     if st.button("Find Similar Prompts"):
@@ -33,8 +42,13 @@ def main():
                     if similar_prompts:
                         st.subheader("Similar Prompts:")
                         for item in similar_prompts:
-                            st.write(f"**Score:** {item['score']:.2f}")
-                            st.write(f"**Prompt:** {item['prompt']}")
                             st.write("---")
                     else:
                         st.write("No similar prompts found.")

 import requests
 # Define the URL of the FastAPI service
+API_URL = "https://lazarr19-prompt-engine.hf.space/most_similar"  # Adjust if the FastAPI service is hosted elsewhere
 def get_similar_prompts(query, n):
         return None
+def get_color(score):
+    if score >= 0.8:
+        return "green"
+    elif score >= 0.5:
+        return "orange"
+    else:
+        return "red"
 def main():
     st.title("Prompt Similarity Finder")
     # User input for query
     query = st.text_input("Enter your query:", "")
     n = st.slider(
+        "Number of similar prompts to retrieve:", min_value=1, max_value=40, value=5
     )
     if st.button("Find Similar Prompts"):
                     if similar_prompts:
                         st.subheader("Similar Prompts:")
                         for item in similar_prompts:
+                            score = item["score"]
+                            color = get_color(score)
+                            # Apply color only to the score part
+                            st.markdown(
+                                f"<p><strong>Score:</strong> <span style='color:{color};'>{score:.2f}</span> <br> <strong>Prompt:</strong> {item['prompt']}</p>",
+                                unsafe_allow_html=True,
+                            )
                             st.write("---")
                     else:
                         st.write("No similar prompts found.")

main.py CHANGED Viewed

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from typing import List
 from src.search_engine import PromptSearchEngine
@@ -47,3 +48,36 @@ async def get_most_similar(query_request: QueryRequest):
         return response
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))

 from fastapi import FastAPI, HTTPException
+from fastapi.responses import HTMLResponse
 from pydantic import BaseModel
 from typing import List
 from src.search_engine import PromptSearchEngine
         return response
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
+@app.get("/", response_class=HTMLResponse)
+async def home_page():
+    return HTMLResponse(
+        """
+        <!DOCTYPE html>
+        <html lang="en">
+        <head>
+            <meta charset="UTF-8">
+            <meta name="viewport" content="width=device-width, initial-scale=1.0">
+            <title>Prompt Search Engine</title>
+            <style>
+                body { font-family: Arial, sans-serif; margin: 20px; }
+                h1 { color: #333; }
+                p { margin-bottom: 10px; }
+                code { background: #f4f4f4; padding: 2px 4px; border-radius: 4px; }
+                .container { max-width: 800px; margin: 0 auto; }
+            </style>
+        </head>
+        <body>
+            <div class="container">
+                <h1>Prompt Search Engine API</h1>
+                <p>Use this API to find similar prompts based on a query.</p>
+                <h2>POST /most_similar</h2>
+                <p><strong>Request:</strong> <code>{"query": "string", "n": 5}</code></p>
+                <p><strong>Response:</strong> <code>{"similar_prompts": [{"score": 0.95, "prompt": "Example prompt 1"}]}</code></p>
+                <p>For more info, visit <a href="https://github.com/your-repository">GitHub</a>.</p>
+            </div>
+        </body>
+        </html>
+        """
+    )

poe/common-tasks.toml CHANGED Viewed

@@ -52,5 +52,14 @@ help     = "Run all checks on the code base"
 sequence = ["style", "types", "lint", "clean"]
 [tool.poe.tasks.ui]
-help    = "Start the UI"
-cmd     = "streamlit run ./frontend/app_ui.py"

 sequence = ["style", "types", "lint", "clean"]
 [tool.poe.tasks.ui]
+help = "Start the UI"
+cmd  = "streamlit run ./frontend/app_ui.py"
+[tool.poe.tasks.build]
+help = "Build the Docker image"
+cmd  =  "docker build -t prompt-search-engine ."
+[tool.poe.tasks.start]
+help = "Run the Docker container"
+cmd  =  "docker run -d -p 8000:8000 prompt-serach-engine"