initial setup - part2

app.py

@@ -1 +1,24 @@
-import gradio 
+import gradio as gr 
+from typing import List, Dict
+import json
+
+def query_tool_similarity(query: str, tools_json: str, top_k: int):
+    """Calculates similarity between query and each tool. 
+        Returns list of int as index for each tool in based on similarity."""
+    try:
+        tools = json.loads(tools_json)
+        assert isinstance(tools, list) and all(isinstance(d, dict) for d in tools)
+    except Exception as e:
+        return f"Invalid JSON for tools: {e}"
+    return f"{query}, {tools}, {top_k}"
+
+with gr.Blocks() as demo: 
+    gr.Interface(fn = query_tool_similarity, 
+                 inputs= [gr.Textbox(label="Query"), 
+                          gr.Textbox(label="Tools (List of Dicts as JSON)", lines=6, placeholder='e.g. [{"name": "foo", "desc": "bar"}]'), 
+                          gr.Number(label="Top K", precision=0)],
+                 outputs= gr.Textbox(label="Result")
+                 )
+
+if __name__ == "__main__":
+    demo.launch()