Update app.py

app.py

@@ -43,13 +43,17 @@ def get_code(query):
                     truncate="LEFT").embeddings
 
     # Retrieve the nearest neighbors
-    similar_item_ids = search_index.get_nns_by_vector(query_embed[0],3,
-                                                    include_distances=True)
+    similar_item_ids = search_index.get_nns_by_vector(query_embed[0],1)
     
-    return data_df.iloc[similar_item_ids[0]]['function_body'] , data_df.iloc[similar_item_ids[0]]['file_path']
-
-iface = gr.Interface(fn=get_code, inputs="text", outputs=[gr.Markdown(), gr.Textbox()])
-iface.launch(debug=True)
+   
+    return data_df.iloc[similar_item_ids[0]]['function_body'], data_df.iloc[similar_item_ids[0]]['file_path']
+examples = ['compute diffusion of given data']
+inputs = gr.Textbox(label='query')
+outputs = [gr.Textbox(label='matched function'), gr.Textbox(label='File path')
+title = "Search Code"
+description = "Semantically search codebase using Cohere embed API. This demo uses Open AI point cloud codebase as an example"
+iface = gr.Interface(fn=get_code, inputs=inputs, outputs=outputs, description = description)
+iface.launch()