Update app.py

app.py

@@ -6,11 +6,13 @@ from google.genai import types
 import requests
 from cryptography.fernet import Fernet
 import os
+import uuid
 
 Url = os.getenv('URL')
 Api_key = os.getenv('API_KEY')
 Key = os.getenv('KEY')
 System_instruction = os.getenv('System_instruction')
+AGENTS_MD_CONTENT = os.getenv('AGENTS_MD_CONTENT')
 
 client = genai.Client(api_key=Api_key)
 cipher = Fernet(Key.encode())
@@ -24,16 +26,17 @@ class InputPrompt(BaseModel):
 
 @app.post("/optimize")
 async def optimize_text(prompt: InputPrompt):
+    unique_id = str(uuid.uuid4())
     optimized_text = gen(prompt.input_prompt)
     url = Url
     data = {
+        "id": unique_id,
         "a": prompt.input_prompt,
         "b": optimized_text
     }
-    print(optimize_text)
     encrypted_data = {k: cipher.encrypt(v.encode()).decode() for k, v in data.items()}
     response = requests.post(url, json=encrypted_data)
-    return {"optimized_text": optimized_text}
+    return {"id": unique_id, "optimized_text": optimized_text}
 
 def gen(prompt):
     try:
@@ -53,40 +56,12 @@ def gen(prompt):
         raise HTTPException(status_code=500, detail="AI Generation Failed")
 
 
-
-
-AGENTS_MD_CONTENT = """# AI Prompt Optimizer Agent
-
-An asynchronous FastAPI microservice that optimizes input prompts using the Google GenAI SDK (`gemma-4-26b-a4b-it`).
-
----
-
-## API Endpoints
-
-### Optimize Prompt
-
-* **Endpoint:** `/optimize`
-* **Method:** `POST`
-* **Content-Type:** `application/json`
-
-#### Request Body
-```json
-{"input_prompt": "Your text or prompt to optimize here."}
-
-```
-
-#### Response Body
-
-```json
-{"optimized_text": "The optimized result string."}
-
-```
-
----
-
-## Error Handling
-
-* **500 Internal Server Error:** Raised with detail `AI Generation Failed` if the GenAI SDK encounters communication issues, API limits, or parsing exceptions."""
+@app.post("/vote")
+async def vote(unique_id: str, is_upvote: bool):
+    data = {"id": unique_id, "vote": str(is_upvote)}
+    response = requests.post(f"{Url}/vote", json=data)
+    return {"status": "success"}
+    
 
 @app.get("/agents.md")
 async def get_agents_md():
@@ -100,6 +75,7 @@ async def get_robots_txt():
     "Allow: /\n"
     "Allow: /agents.md\n"
     "Disallow: /optimize\n"
+    "Disallow: /vote\n"
     )
     return Response(content=robots_content, media_type="text/plain")
 
@@ -118,6 +94,7 @@ async def read_items():
   <meta name="description" content="Got a messy prompt? Paste it in and Prompt Optimizer rewrites it into something clear and specific in seconds — no account, no templates to fill out, no signup. Works for ChatGPT, Claude, Gemini, and any AI model." />
   <meta name="author" content="Prompt Optimizer Bot" />
   <meta name="robots" content="index, follow" />
+  <link rel="canonical" href="https://codebyam-prompt-optimizer.hf.space/" />
   <meta name="theme-color" content="#3D5AFE" />
 
   <!-- Open Graph / Facebook -->
@@ -557,6 +534,47 @@ async def read_items():
       transform: translate(1px, 1px);
       box-shadow: 1px 1px 0 var(--ink);
     }
+		.output-card .action-row {
+			display: flex;
+			align-items: center;
+			gap: 8px;
+			margin-top: 12px;
+		}
+		.output-card .copy-btn {
+			margin-top: 0; /* moved spacing to .action-row */
+		}
+		.vote-btn {
+			background-color: white;
+			color: var(--ink);
+			border: 2px solid var(--ink);
+			border-radius: 8px;
+			padding: 6px 10px;
+			font-size: 0.9em;
+			cursor: pointer;
+			box-shadow: 2px 2px 0 var(--ink);
+			transition: transform 0.12s ease, box-shadow 0.12s ease, background-color 0.12s ease;
+			line-height: 1;
+		}
+		.vote-btn:hover {
+			transform: translate(-1px, -1px);
+			box-shadow: 3px 3px 0 var(--ink);
+		}
+		.vote-btn:active {
+			transform: translate(1px, 1px);
+			box-shadow: 1px 1px 0 var(--ink);
+		}
+		.vote-btn.selected-up {
+			background-color: var(--green);
+			color: white;
+		}
+		.vote-btn.selected-down {
+			background-color: var(--coral);
+			color: white;
+		}
+		.vote-btn:disabled {
+			opacity: 0.55;
+			cursor: default;
+		}
 
     .input-bar {
       display: flex;
@@ -842,35 +860,85 @@ async def read_items():
         }
 
         if (role === "bot" && withCopy) {
-          // Deliverable-style output card
-          const card = document.createElement("div");
-          card.classList.add("output-card");
-
-          const tag = document.createElement("div");
-          tag.classList.add("sticky-tag");
-          tag.textContent = "Refined Prompt ✨";
-
-          const bodyDiv = document.createElement("div");
-          bodyDiv.classList.add("card-body");
-          bodyDiv.innerHTML = formatTextWithLineBreaks(text);
-
-          const copyBtn = document.createElement("button");
-          copyBtn.classList.add("copy-btn");
-          copyBtn.textContent = "Copy";
-          copyBtn.addEventListener("click", () => {
-            navigator.clipboard.writeText(text);
-            copyBtn.textContent = "Copied!";
-            setTimeout(() => (copyBtn.textContent = "Copy"), 1500);
-          });
-
-          card.appendChild(tag);
-          bodyDiv.appendChild(document.createElement("br"));
-          bodyDiv.appendChild(copyBtn);
-          card.appendChild(bodyDiv);
-
-          msgDiv.appendChild(avatar);
-          msgDiv.appendChild(card);
-        } else {
+					// Deliverable-style output card
+					const card = document.createElement("div");
+					card.classList.add("output-card");
+					if (options.uniqueId) card.dataset.uniqueId = options.uniqueId;
+
+					const tag = document.createElement("div");
+					tag.classList.add("sticky-tag");
+					tag.textContent = "Refined Prompt ✨";
+
+					const bodyDiv = document.createElement("div");
+					bodyDiv.classList.add("card-body");
+					bodyDiv.innerHTML = formatTextWithLineBreaks(text);
+
+					const actionRow = document.createElement("div");
+					actionRow.classList.add("action-row");
+
+					const copyBtn = document.createElement("button");
+					copyBtn.classList.add("copy-btn");
+					copyBtn.textContent = "Copy";
+					copyBtn.addEventListener("click", () => {
+						navigator.clipboard.writeText(text);
+						copyBtn.textContent = "Copied!";
+						setTimeout(() => (copyBtn.textContent = "Copy"), 1500);
+					});
+
+					const upBtn = document.createElement("button");
+					upBtn.classList.add("vote-btn");
+					upBtn.textContent = "👍";
+					upBtn.setAttribute("aria-label", "Upvote this result");
+
+					const downBtn = document.createElement("button");
+					downBtn.classList.add("vote-btn");
+					downBtn.textContent = "👎";
+					downBtn.setAttribute("aria-label", "Downvote this result");
+
+					// restore prior vote state if reloaded from history
+					if (options.voted === "up") upBtn.classList.add("selected-up");
+					if (options.voted === "down") downBtn.classList.add("selected-down");
+					if (options.voted) { upBtn.disabled = true; downBtn.disabled = true; }
+
+					async function castVote(isUpvote, chosenBtn, otherBtn) {
+						const uid = card.dataset.uniqueId;
+						if (!uid) return; // no id, nothing to send
+						upBtn.disabled = true;
+						downBtn.disabled = true;
+						chosenBtn.classList.add(isUpvote ? "selected-up" : "selected-down");
+						card.dataset.voted = isUpvote ? "up" : "down";
+						try {
+							await fetch("vote", {
+								method: "POST",
+								headers: { "Content-Type": "application/json" },
+								body: JSON.stringify({ unique_id: uid, is_upvote: isUpvote }),
+							});
+							await saveHistory();
+						} catch (err) {
+							console.warn("⚠️ Failed to record vote:", err);
+							// allow retry on failure
+							upBtn.disabled = false;
+							downBtn.disabled = false;
+							chosenBtn.classList.remove(isUpvote ? "selected-up" : "selected-down");
+							delete card.dataset.voted;
+						}
+					}
+
+					upBtn.addEventListener("click", () => castVote(true, upBtn, downBtn));
+					downBtn.addEventListener("click", () => castVote(false, downBtn, upBtn));
+
+					actionRow.appendChild(copyBtn);
+					actionRow.appendChild(upBtn);
+					actionRow.appendChild(downBtn);
+
+					card.appendChild(tag);
+					bodyDiv.appendChild(document.createElement("br"));
+					bodyDiv.appendChild(actionRow);
+					card.appendChild(bodyDiv);
+
+					msgDiv.appendChild(avatar);
+					msgDiv.appendChild(card);
+				} else {
           const bubble = document.createElement("div");
           bubble.classList.add("bubble");
           if (isLoading) {
@@ -893,33 +961,40 @@ async def read_items():
         if (loadingMsg) chatWindow.removeChild(loadingMsg);
       }
 
-      async function saveHistory() {
-        const messages = [];
-        chatWindow.querySelectorAll(".message").forEach((msg) => {
-          const role = msg.classList.contains("user-message") ? "user" : "bot";
-          const textEl = msg.querySelector(".card-body") || msg.querySelector(".bubble");
-          const text = textEl?.innerText || "";
-          const withCopy = !!msg.querySelector(".output-card");
-          messages.push({ role, text, withCopy });
-        });
-        try {
-          localStorage.setItem(STORAGE_KEY, JSON.stringify(messages));
-        } catch (err) {
-          console.warn("⚠️ Failed to save chat history:", err);
-        }
-      }
-
-      async function loadHistory() {
-        try {
-          const result = localStorage.getItem(STORAGE_KEY);
-          const messages = result ? JSON.parse(result) : [];
-          messages.forEach((msg) =>
-            appendMessage(msg.role, msg.text, { withCopy: msg.withCopy ?? (msg.role === "bot") })
-          );
-        } catch (err) {
-          console.warn("⚠️ Failed to load chat history:", err);
-        }
-      }
+			async function saveHistory() {
+				const messages = [];
+				chatWindow.querySelectorAll(".message").forEach((msg) => {
+					const role = msg.classList.contains("user-message") ? "user" : "bot";
+					const textEl = msg.querySelector(".card-body") || msg.querySelector(".bubble");
+					const text = textEl?.innerText || "";
+					const withCopy = !!msg.querySelector(".output-card");
+					const cardEl = msg.querySelector(".output-card");
+					const uniqueId = cardEl?.dataset.uniqueId || null;
+					const voted = cardEl?.dataset.voted || null;
+					messages.push({ role, text, withCopy, uniqueId, voted });
+				});
+				try {
+					localStorage.setItem(STORAGE_KEY, JSON.stringify(messages));
+				} catch (err) {
+					console.warn("⚠️ Failed to save chat history:", err);
+				}
+			}
+
+			async function loadHistory() {
+				try {
+					const result = localStorage.getItem(STORAGE_KEY);
+					const messages = result ? JSON.parse(result) : [];
+					messages.forEach((msg) =>
+						appendMessage(msg.role, msg.text, {
+							withCopy: msg.withCopy ?? (msg.role === "bot"),
+							uniqueId: msg.uniqueId,
+							voted: msg.voted,
+						})
+					);
+				} catch (err) {
+					console.warn("⚠️ Failed to load chat history:", err);
+				}
+			}
 
       clearBtn.addEventListener("click", async () => {
         try { localStorage.removeItem(STORAGE_KEY); } catch (e) {}
@@ -930,42 +1005,43 @@ async def read_items():
           </div>`;
       });
 
-      async function handleSend() {
-        const prompt = userInput.value.trim();
-        if (!prompt) return;
-        appendMessage("user", prompt);
-        userInput.value = "";
-        userInput.style.height = "auto";
-        appendMessage("bot", "", { isLoading: true });
-        await saveHistory();
-        try {
-          const response = await fetch(API_ENDPOINT, {
-            method: "POST",
-            headers: { "Content-Type": "application/json" },
-            body: JSON.stringify({ input_prompt: prompt }),
-          });
-          removeLoadingMessage();
-          if (!response.ok) throw new Error(`API Error: ${response.status} ${response.statusText}`);
-          const data = await response.json();
-          const optimized = data.optimized_text || data.result || data.output || null;
-          if (optimized) {
-            appendMessage("bot", optimized, { withCopy: true });
-          } else {
-            appendMessage("bot", "⚠️ The API responded successfully but didn't return optimized text.");
-          }
-          await saveHistory();
-        } catch (error) {
-          console.error("Optimization failed:", error);
-          removeLoadingMessage();
-          appendMessage(
-            "bot",
-            `🚨 Connection error: ${error.message}
+      async function handleSend() {
+        const prompt = userInput.value.trim();
+        if (!prompt) return;
+        appendMessage("user", prompt);
+        userInput.value = "";
+        userInput.style.height = "auto";
+        appendMessage("bot", "", { isLoading: true });
+        await saveHistory();
+        try {
+          const response = await fetch(API_ENDPOINT, {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+            body: JSON.stringify({ input_prompt: prompt }),
+          });
+          removeLoadingMessage();
+          if (!response.ok) throw new Error(`API Error: ${response.status} ${response.statusText}`);
+          const data = await response.json();
+					const optimized = data.optimized_text || data.result || data.output || null;
+					if (optimized) {
+						appendMessage("bot", optimized, { withCopy: true, uniqueId: data.id });
+					} else {
+						appendMessage("bot", "⚠️ The API responded successfully but didn't return optimized text.");
+					}
+					await saveHistory();
+        } catch (error) {
+          console.error("Optimization failed:", error);
+          removeLoadingMessage();
+          appendMessage(
+            "bot",
+            `🚨 Connection error: ${error.message}
 
 💡 Tip: Make sure your optimization API is running and accessible.`
-          );
-          await saveHistory();
-        }
-      }
+          );
+          await saveHistory();
+        }
+      }
+
 
       sendButton.addEventListener("click", handleSend);
       userInput.addEventListener("input", () => {