Update app.py

app.py

@@ -1,6 +1,3 @@
-import gradio as gr
-from gradio_client import Client
-
 
 title = """# 🙋🏻‍♂️Welcome to tonic's e5 connector for e5Mistral7b
 ⚒️this connector looks like openai embeddings but uses e5Mistral7b - so you can use it as a drop in replacement for the open ai api ⚒️
@@ -19,10 +16,27 @@ def get_embeddings(task, input_text):
     except Exception as e:
         return str(e)
 
-### connector ;-)
+def format_response(embeddings, model):
+    return {
+        "data": [
+            {
+                "embedding": embeddings,
+                "index": 0,
+                "object": "embedding"
+            }
+        ],
+        "model": model,
+        "object": "list",
+        "usage": {
+            "prompt_tokens": 17
+            "total_tokens": 17
+        }
+    }
+
 def generate_embeddings(input_text, model, encoding_format, user):
     embeddings = get_embeddings(model, input_text)
-    return embeddings
+    formatted_response = format_response(embeddings, model)
+    return formatted_response
 
 with gr.Blocks() as app:
     gr.Markdown(title)
@@ -33,7 +47,7 @@ with gr.Blocks() as app:
         user = gr.Textbox(label="User", placeholder="Enter user identifier (optional)")
         submit_button = gr.Button("Generate Embeddings")
 
-    output = gr.Textbox(label="Embeddings Output", lines=10)
+    output = gr.JSON(label="Embeddings Output")
 
     submit_button.click(
         fn=generate_embeddings,