Update app.py

app.py

@@ -141,6 +141,27 @@ def search_similar_sentences(input_question, corpus_sentences, corpus_embeddings
     results = [(corpus_sentences[i], top_k_scores[i]) for i in top_k_indices]
     return results
 
+# openai response object formatting
+def format_response(embeddings):
+    return {
+        "data": [
+            {
+                "embedding": embeddings,
+                "index": 0,
+                "object": "embedding"
+            }
+        ],
+        "model": "e5-mistral",
+        "object": "list",
+        "usage": {
+            "prompt_tokens": 17,
+            "total_tokens": 17
+        }
+    }
+
+def generate_and_format_embeddings(selected_task, input_text):
+    embeddings = compute_embeddings(selected_task, input_text)
+    return format_response(embeddings)
 
 def app_interface():
     corpus_sentences = []
@@ -211,6 +232,22 @@ def app_interface():
                 outputs=search_results_output
             )
 
+        with gr.Tab("Connector-like Embeddings"):
+            with gr.Row():
+                input_text_box_connector = gr.Textbox(label="Input Text", placeholder="Enter text or array of texts")
+                model_dropdown_connector = gr.Dropdown(label="Model", choices=["ArguAna", "ClimateFEVER", "DBPedia", "FEVER", "FiQA2018", "HotpotQA", "MSMARCO", "NFCorpus", "NQ", "QuoraRetrieval", "SCIDOCS", "SciFact", "Touche2020", "TRECCOVID"], value="text-embedding-ada-002")
+                encoding_format_connector = gr.Radio(label="Encoding Format", choices=["float", "base64"], value="float")
+                user_connector = gr.Textbox(label="User", placeholder="Enter user identifier (optional)")
+                submit_button_connector = gr.Button("Generate Embeddings")
+
+            output_display_connector = gr.JSON(label="Embeddings Output")
+
+            submit_button_connector.click(
+                fn=generate_and_format_embeddings,
+                inputs=[input_text_box_connector, model_dropdown_connector, encoding_format_connector, user_connector],
+                outputs=output_display_connector
+            )
+
         with gr.Row():
             with gr.Column():
                 input_text_box