Update app.py

app.py

@@ -13,7 +13,7 @@ model = sentence_transformers.SentenceTransformer(
 dataset = datasets.load_dataset("json", data_files=["./dataset.json"], split="train")
 dataset.load_faiss_index("embeddings", "index.faiss")
 
-def search(query, k):
+def search(query, k=3):
     query_embedding = model.encode(query)
     _, retrieved_examples = dataset.get_nearest_examples(
         "embeddings",
@@ -38,21 +38,54 @@ def search(query, k):
         results.append(result)
     return results
 
+# Add custom styles to the interface
+styles = gr.Styles(
+    button_color="white",
+    label_font="Arial",
+    label_font_size=18,
+    label_color="black",
+    input_font="Arial",
+    input_font_size=18,
+    input_color="black",
+    output_font="Arial",
+    output_font_size=18,
+    output_color="black",
+    full_width=True,
+    background="#f2f2f2",
+)
+
 iface = gr.Interface(
-    search,
+    fn=search,
     inputs=[
-        gr.inputs.Textbox(label="Query"),
-        gr.inputs.Number(label="K", default=3),
+        gr.inputs.Textbox(
+            label="Query", placeholder="Type in a search query...", lines=3
+        ),
+        gr.inputs.Number(
+            label="K",
+            default=3,
+            description="Number of results to return",
+        ),
     ],
     outputs=[
-        gr.outputs.Textbox(label="Title"),
-        gr.outputs.Textbox(label="Transcript"),
-        gr.outputs.Textbox(label="Link"),
+        gr.outputs.Label(
+            label="Result 1", type="auto", default="Search results will appear here."
+        ),
+        gr.outputs.Label(
+            label="Result 2", type="auto", default=""
+        ),
+        gr.outputs.Link(
+            label="Result 3", type="auto", default=""
+        ),
     ],
     title="Camembert and Faiss-powered Search Engine",
     description="Search through a dataset using Camembert and Faiss",
-    theme="light",
+    theme="default",
     layout="vertical",
+    allow_flagging=False,
+    allow_screenshot=False,
+    allow_share=True,
+    allow_download=False,
+    style=styles,
 )
 
 iface.launch()