Update app.py

app.py

@@ -92,6 +92,121 @@ def get_data(query, top_k, score, db_col, db_index):
 
     return jsons
 
+def run_command(command):
+    try:
+        result = subprocess.check_output(command, shell=True, text=True)
+        return result
+    except subprocess.CalledProcessError as e:
+        return f"Error: {e}"
+
+
+with gr.Blocks(
+    title="3GPP Database",
+    theme="Base",
+    css=""".bigbox {
+    min-height:250px;
+}"""
+) as demo:
+    with gr.Tab("Matching"):
+        with gr.Accordion("Vector similarity"):
+            with gr.Row():
+                with gr.Column():
+                    top_k = gr.Slider(1,
+                                      TOP_K_MAX,
+                                      value=TOP_K_DEFAULT,
+                                      step=1,
+                                      label="Vector similarity top_k",
+                                      interactive=True)
+                with gr.Column():
+                    score = gr.Slider(0.01,
+                                      0.99,
+                                      value=SCORE_DEFAULT,
+                                      step=0.01,
+                                      label="Vector similarity score",
+                                      interactive=True)
+
+        with gr.Row():
+            with gr.Column(scale=10):
+                input_box = gr.Textbox(label="Input", placeholder="What are you looking for?")
+            with gr.Column(scale=1, min_width=BUTTON_MIN_WIDTH):
+                btn_run = gr.Button("Run", variant="primary")
+
+        output_box = gr.JSON(label="Output")
+
+        btn_run.click(fn=get_data,
+                      inputs=[input_box, top_k, score, db_col_textbox, db_index_textbox],
+                      outputs=[output_box])
+
+    with gr.Tab("Configuration"):
+        with gr.Row():
+            btn_init = gr.Button("Init")
+
+        load_emb = gr.Textbox("", label='Embedding Client', show_label=True)
+        load_col = gr.Textbox("", label='Milvus Collection', show_label=True)
+
+        with gr.Accordion("Embedding"):
+
+            with gr.Row():
+                with gr.Column():
+                    emb_textbox = gr.Textbox(
+                        label="Embedding Model",
+                        value=EMBEDDING_MODEL,
+                        placeholder="Paste Your Embedding Model Repo on HuggingFace",
+                        lines=1,
+                        interactive=True,
+                        type='email')
+
+                with gr.Column():
+                    emb_dropdown = gr.Dropdown(
+                        EMBEDDING_LIST,
+                        value=EMBEDDING_LOADER,
+                        multiselect=False,
+                        interactive=True,
+                        label="Embedding Loader")
+
+        with gr.Accordion("Milvus Database"):
+            with gr.Row():
+                db_col_textbox = gr.Textbox(
+                    label="Milvus Collection",
+                    value=MILVUS_COLLECTION,
+                    placeholder="Paste Your Milvus Collection (xx-xx-xx) and Hit ENTER",
+                    lines=1,
+                    interactive=True,
+                    type='email')
+                db_index_textbox = gr.Textbox(
+                    label="Milvus Host",
+                    value=MILVUS_HOST,
+                    placeholder="Paste Your Milvus Index (xxxx) and Hit ENTER",
+                    lines=1,
+                    interactive=True,
+                    type='password')
+
+    with gr.Tab("Command Output Viewer"):
+        with gr.Row():
+            cmd_input = gr.Textbox(label="Command", placeholder="Enter a command")
+            cmd_btn_run = gr.Button("Run", variant="primary")
+        cmd_output = gr.Textbox(label="Output", placeholder="Command output will be displayed here", readonly=True)
+
+        cmd_iface = gr.Interface(
+            fn=run_command,
+            inputs="text",
+            outputs="text",
+            title="Command Output Viewer",
+            description="Enter a command and view its output.",
+            examples=[
+                ["ls"],
+                ["pwd"],
+                ["echo 'Hello, Gradio!'"],
+                ["python --version"]]
+        )
+
+        cmd_btn_run.click(fn=cmd_iface.launch, inputs=[cmd_input], outputs=[cmd_output])
+
+if __name__ == "__main__":
+    demo.queue()
+    demo.launch(server_name="0.0.0.0", server_port=7860)
+
+'''
 with gr.Blocks(
     title = "3GPP Database",
     theme = "Base",
@@ -173,6 +288,9 @@ with gr.Blocks(
                     interactive=True,
                     type='password')
 
+
+                
+
     btn_init.click(fn=init_emb,
                    inputs=[emb_textbox, emb_dropdown, db_col_textbox],
                    outputs=[load_emb, load_col])
@@ -185,3 +303,4 @@ if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0",
                 server_port=7860)
 
+'''