Update app.py

app.py

@@ -161,14 +161,14 @@ def classify_prompt(prompt: str, threshold: float) -> Tuple[dict, gr.DataFrame]:
         # res = f"This prompt looks malicious, here are some similar threats:\n{json.dumps(asdict(classification), indent=4)}"
         return classification, gr.DataFrame(
             [
-                (r.known_prompt, r.similarity_percentage, r.source)
+                (r.similarity_percentage, r.known_prompt, r.source)
                 for r in classification.reason
             ]
         )
 
     res = (
         f"{classification} - This prompt is not similar to any learned prompt above {int(threshold*100)}%, Try a lower threshold.",
-        gr.DataFrame([("No similar prompt found", 0.0, "N/A")]),
+        gr.DataFrame([(0.0, "No similar prompt found", "N/A")]),
     )
     return res
 
@@ -198,7 +198,7 @@ iface = gr.Interface(
     outputs=[
         "text",
         gr.Dataframe(
-            headers=["Prompt", "Similarity", "Source"],
+            headers=["Similarity", "Prompt", "Source"],
             datatype=["str", "number", "str"],
             row_count=1,
             col_count=(3, "fixed"),