Update app.py

app.py

@@ -50,6 +50,7 @@ with zipfile.ZipFile("artifacts/unimoda.zip", 'r') as zip_ref:
     zip_ref.extractall(RAW_PHOTOS_DIR)
 
 
+
 class TextEncoder:
     """Encodes the given text"""
 
@@ -121,37 +122,54 @@ def main(args):
     )
 
 
-def make_frontend(
-    fn: Callable[[Image], str], flagging: bool = False, gantry: bool = False, app_name: str = "fashion-aggregator"
-):
+def make_frontend(fn: Callable[[str], List[str]], app_name: str = "fashion-aggregator"):
     """Creates a gradio.Interface frontend for text to image search function."""
 
-    allow_flagging = "never"
+    def preprocess_input(input_str):
+        paths_and_scores = fn(input_str)
+        images = [gr.utils.image.load_image(path) for path in paths_and_scores]
+        return images
+
+    input_section = gr.inputs.Textbox(lines=2, placeholder="Enter item description here")
+
+    output_section = gr.Gallery(label="Relevant Items", show_labels=True, label_position="below")
+
+    # Define Interface layout with Gradio's Row and Column
+    interface_layout = gradio.ui.Row(
+        gradio.ui.Column(
+            gradio.ui.Textbox("Fashion Aggregator", label="Title"),
+            gradio.ui.Image(LOGO, label="Logo"),
+            gradio.ui.Textbox("Discover your perfect apparel effortlessly. Simply describe what you're looking for!", label="Description"),
+        ),
+        gradio.ui.Column(
+            gradio.ui.Section(
+                gradio.ui.Center(
+                    input_section,
+                ),
+                gradio.ui.Center(
+                    output_section,
+                ),
+            )
+        ),
+        gradio.ui.Column(
+            gradio.ui.Textbox("Disclaimer: The search results are based on embeddings and may not be always accurate.", label="Disclaimer"),
+        ),
+    )
 
-    # HTML code to include the logo
-    logo_html = f"<img src='path_to_your_logo' alt='Logo' style='width: 100px; height: 100px;' />"
-    
     # Build a customized browser interface to a Python function
-    layout = gr.Layout(
-        gr.Row(
-            [
-                gr.Column([gr.outputs.HTML(logo_html), gr.Interface(
-                    fn=fn,
-                    outputs=gr.Gallery(label="Relevant Items", show_labels=True, label_position="below"),
-                    inputs=gr.inputs.Textbox(label="Item Description", placeholder="Enter item description here"),
-                    title="Fashion Aggregator",
-                    thumbnail=LOGO,
-                    description="Discover your perfect apparel effortlessly. Simply describe what you're looking for!",
-                    cache_examples=False,
-                    allow_flagging=allow_flagging,
-                    flagging_options=["incorrect", "offensive", "other"],
-                )]),
-                gr.Column([gr.outputs.HTML("This web app is for demonstration purposes only and not intended for commercial use.")])
-            ]
-        )
+    frontend = gr.Interface(
+        fn=preprocess_input,
+        inputs=input_section,
+        outputs=output_section,
+        layout=interface_layout,
+        allow_flagging=False,
+        title=app_name,
+        thumbnail=LOGO,
+        description="Enter the description of the item you're looking for and see the corresponding images!",
     )
 
-    return layout
+    return frontend
+