Update app.py

app.py

@@ -8,19 +8,19 @@ model = AutoModel.from_pretrained("jinaai/jina-clip-v1", trust_remote_code=True)
 
 # Function to compute similarity
 def compute_similarity(input1, input2, input1_type, input2_type):
-    # Check if inputs are empty
-    if input1_type == "Text" and not input1.strip():
+    # Ensure inputs are valid
+    if input1_type == "Text" and (not input1 or input1.strip() == ""):
         return "Error: Input 1 is empty!"
-    if input2_type == "Text" and not input2.strip():
+    if input2_type == "Text" and (not input2 or input2.strip() == ""):
         return "Error: Input 2 is empty!"
     if input1_type == "Image" and input1 is None:
         return "Error: Image 1 is missing!"
     if input2_type == "Image" and input2 is None:
         return "Error: Image 2 is missing!"
 
+    # Encode inputs
     inputs = []
-
-    # Process first input
+    
     if input1_type == "Text":
         text1_embedding = model.encode_text([input1])
         inputs.append(text1_embedding)
@@ -28,7 +28,6 @@ def compute_similarity(input1, input2, input1_type, input2_type):
         image1_embedding = model.encode_image([Image.fromarray(input1)])
         inputs.append(image1_embedding)
 
-    # Process second input
     if input2_type == "Text":
         text2_embedding = model.encode_text([input2])
         inputs.append(text2_embedding)
@@ -38,16 +37,15 @@ def compute_similarity(input1, input2, input1_type, input2_type):
 
     # Compute cosine similarity
     similarity_score = (inputs[0] @ inputs[1].T).item()
+    return f"Similarity Score: {similarity_score:.4f}"
 
-    return similarity_score
-
-# Function to update UI based on selected input types
+# Function to toggle input fields dynamically
 def update_visibility(input1_type, input2_type):
     return (
-        gr.update(visible=input1_type == "Text", value="" if input1_type == "Text" else None),
-        gr.update(visible=input1_type == "Image", value=None if input1_type == "Image" else None),
-        gr.update(visible=input2_type == "Text", value="" if input2_type == "Text" else None),
-        gr.update(visible=input2_type == "Image", value=None if input2_type == "Image" else None),
+        gr.update(visible=(input1_type == "Text")),  # Show text input if Text is selected
+        gr.update(visible=(input1_type == "Image")), # Show image input if Image is selected
+        gr.update(visible=(input2_type == "Text")),  
+        gr.update(visible=(input2_type == "Image"))
     )
 
 # Gradio UI
@@ -59,19 +57,22 @@ with gr.Blocks() as demo:
         input2_type = gr.Radio(["Text", "Image"], label="Input 2 Type", value="Image")
 
     with gr.Row():
-        input1 = gr.Textbox(label="Text Input 1", visible=True)
-        image1 = gr.Image(type="numpy", label="Image Input 1", visible=False)
-    
+        input1_text = gr.Textbox(label="Text Input 1", visible=True)
+        input1_image = gr.Image(type="numpy", label="Image Input 1", visible=False)
+
     with gr.Row():
-        input2 = gr.Textbox(label="Text Input 2", visible=False)
-        image2 = gr.Image(type="numpy", label="Image Input 2", visible=True)
+        input2_text = gr.Textbox(label="Text Input 2", visible=False)
+        input2_image = gr.Image(type="numpy", label="Image Input 2", visible=True)
 
     output = gr.Textbox(label="Similarity Score / Error", interactive=False)
 
-    input1_type.change(update_visibility, inputs=[input1_type, input2_type], outputs=[input1, image1, input2, image2])
-    input2_type.change(update_visibility, inputs=[input1_type, input2_type], outputs=[input1, image1, input2, image2])
+    # Toggle visibility of inputs dynamically
+    input1_type.change(update_visibility, inputs=[input1_type, input2_type], 
+                       outputs=[input1_text, input1_image, input2_text, input2_image])
+    input2_type.change(update_visibility, inputs=[input1_type, input2_type], 
+                       outputs=[input1_text, input1_image, input2_text, input2_image])
 
     btn = gr.Button("Compute Similarity")
-    btn.click(compute_similarity, inputs=[input1, input2, input1_type, input2_type], outputs=output)
+    btn.click(compute_similarity, inputs=[input1_text, input2_text, input1_type, input2_type], outputs=output)
 
 demo.launch()