Deploy CodeMode via Agent

app.py

@@ -562,8 +562,77 @@ with gr.Blocks(theme=theme, css=css, title="CodeMode - Baseline vs Fine-tuned")
             compare_btn.click(search_comparison, inputs=search_query, outputs=[baseline_results, finetuned_results])
         
         
-        # TAB 3: DEPLOYMENT MONITORING
-        with gr.Tab("3. Deployment Monitoring"):
+        # TAB 3: CODE SIMILARITY SEARCH
+        with gr.Tab("3. Code Similarity Search"):
+            gr.Markdown("### Find Similar Code Snippets")
+            gr.Markdown("Paste a code snippet to find similar code in the database")
+            
+            with gr.Row():
+                with gr.Column():
+                    code_input = gr.Code(label="Paste Code Snippet", language="python", lines=10)
+                    similarity_btn = gr.Button("Find Similar Code", variant="primary")
+                
+                with gr.Column():
+                    gr.Markdown("#### Search Settings")
+                    top_k_slider = gr.Slider(minimum=1, maximum=20, value=5, step=1, label="Number of Results")
+                    model_choice = gr.Radio(["Baseline", "Fine-tuned", "Both"], value="Both", label="Model to Use")
+            
+            gr.Markdown("### Results")
+            
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("#### Baseline Results")
+                    baseline_code_results = gr.Dataframe(
+                        headers=["File", "Similarity", "Code Snippet"],
+                        datatype=["str", "str", "str"],
+                        interactive=False,
+                        wrap=True,
+                        value=[["No search yet", "-", "-"]]
+                    )
+                
+                with gr.Column():
+                    gr.Markdown("#### Fine-tuned Results")
+                    finetuned_code_results = gr.Dataframe(
+                        headers=["File", "Similarity", "Code Snippet"],
+                        datatype=["str", "str", "str"],
+                        interactive=False,
+                        wrap=True,
+                        value=[["No search yet", "-", "-"]]
+                    )
+            
+            def search_similar_code(code_snippet, top_k, model_choice):
+                if not code_snippet or len(code_snippet.strip()) == 0:
+                    empty = [["Enter code to search", "-", "-"]]
+                    return empty, empty
+                
+                baseline_res = []
+                finetuned_res = []
+                
+                if model_choice in ["Baseline", "Both"]:
+                    baseline_res = search_baseline(code_snippet, top_k)
+                    if not baseline_res:
+                        baseline_res = [["No results found", "-", "-"]]
+                
+                if model_choice in ["Fine-tuned", "Both"]:
+                    finetuned_res = search_finetuned(code_snippet, top_k)
+                    if not finetuned_res:
+                        finetuned_res = [["No results found", "-", "-"]]
+                
+                if model_choice == "Baseline":
+                    finetuned_res = [["Not searched", "-", "-"]]
+                elif model_choice == "Fine-tuned":
+                    baseline_res = [["Not searched", "-", "-"]]
+                
+                return baseline_res, finetuned_res
+            
+            similarity_btn.click(
+                search_similar_code,
+                inputs=[code_input, top_k_slider, model_choice],
+                outputs=[baseline_code_results, finetuned_code_results]
+            )
+        
+        # TAB 4: DEPLOYMENT MONITORING
+        with gr.Tab("4. Deployment Monitoring"):
             gr.Markdown("### Embedding Quality Analysis")
             gr.Markdown("Analyze the semantic space distribution and diversity of embeddings")