Update app.py

app.py

@@ -49,6 +49,7 @@ def embedding_to_text(embedding_tensor):
 # 3. GRADIO UI (Same as before)
 # ==========================================
 
+
 def run_mixing(text1, text2):
     if not text1 or not text2:
         return "Please enter two sentences."
@@ -67,10 +68,33 @@ def run_mixing(text1, text2):
 
     return embedding_to_text(v_mixed)
 
+# Update the mixing function to accept a 'ratio' (0.0 to 1.0)
+def run_weighted_mixing(text1, text2, mix_ratio):
+    if not text1 or not text2:
+        return "Please enter two sentences."
+
+    # 1. Get vectors
+    v1 = text_to_embedding(text1)
+    v2 = text_to_embedding(text2)
+
+    # 2. Align lengths
+    min_len = min(v1.shape[1], v2.shape[1])
+    v1 = v1[:, :min_len, :]
+    v2 = v2[:, :min_len, :]
+
+    # 3. Weighted Average
+    # if ratio is 0.0 -> 100% Text1
+    # if ratio is 1.0 -> 100% Text2
+    # if ratio is 0.5 -> 50/50
+    v_mixed = (v1 * (1 - mix_ratio)) + (v2 * mix_ratio)
+
+    return embedding_to_text(v_mixed)
+
+
 with gr.Blocks(title="FLAN-T5 Latent Explorer", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🧠 FLAN-T5 Latent Space Mixer")
     gr.Markdown(f"Running `{model_name}`. This model is instruction-tuned and much smarter than BART.")
-
+    
     with gr.Row():
         with gr.Column():
             t1 = gr.Textbox(label="Concept A", value="The King is powerful.")
@@ -80,6 +104,21 @@ with gr.Blocks(title="FLAN-T5 Latent Explorer", theme=gr.themes.Soft()) as demo:
             out = gr.Textbox(label="Result", lines=2)
 
     btn.click(run_mixing, inputs=[t1, t2], outputs=out)
+    # Update the mixing function to accept a 'ratio' (0.0 to 1.0)
+
+# In the Gradio UI section:
+with gr.TabItem("2. Vector Mixing"):
+    with gr.Row():
+        t1 = gr.Textbox(label="Start Sentence", value="The dog is happy.")
+        t2 = gr.Textbox(label="End Sentence", value="The cat is angry.")
+    
+    # Add a slider
+    slider = gr.Slider(minimum=0.0, maximum=1.0, value=0.5, step=0.1, label="Mixing Ratio (Left = Start, Right = End)")
+    
+    btn_mix = gr.Button("Morph")
+    out = gr.Textbox(label="Result")
+
+    btn_mix.click(run_weighted_mixing, inputs=[t1, t2, slider], outputs=out)
 
 if __name__ == "__main__":
     demo.launch()