Upload app.py

app.py

@@ -3,34 +3,163 @@ import gradio as gr
 
 with gr.Blocks() as demo:
     gr.Markdown('# vector_search')
-    with gr.Tab('write'):
-        info = gr.Markdown()
+    info = gr.Markdown()
+
+    with gr.Tab('ddg'):
+        ddg_query = gr.Textbox(
+            lines=1,
+            label='Query',
+            interactive=True,
+            show_copy_button=True,
+            )
+        ddg_button = gr.Button(
+            variant='primary',
+            value='Search',
+            )
+        ddg_result = gr.Textbox(
+            lines=20,
+            label='Result',
+            interactive=True,
+            show_copy_button=True,
+            )
+
+    with gr.Tab('bs4'):
+        bs4_url = gr.Textbox(
+            lines=1,
+            label='URL',
+            interactive=True,
+            show_copy_button=True,
+            )
+        bs4_button = gr.Button(
+            variant='primary',
+            value='Scraping',
+            )
+        bs4_result = gr.Textbox(
+            lines=20,
+            label='Result',
+            interactive=True,
+            show_copy_button=True,
+            )
+
+    with gr.Tab('upload'):
+        gr.Markdown('name単位でsearchできます。行ごとにchunkになります。')
+        upload_name = gr.Textbox(
+            lines=1,
+            label='Name',
+            interactive=True,
+            show_copy_button=True,
+            )
+        upload_filename = gr.Textbox(
+            lines=1,
+            label='FileName',
+            interactive=True,
+            show_copy_button=True,
+            )
+        upload_content = gr.Textbox(
+            lines=20,
+            label='Content',
+            interactive=True,
+            show_copy_button=True,
+            )
+        upload_button = gr.Button(
+            variant='primary',
+            value='Upload',
+            )
+
+    with gr.Tab('delete'):
+        delete_name = gr.Textbox(
+            lines=1,
+            label='Name',
+            interactive=True,
+            show_copy_button=True,
+            )
+        delete_filename = gr.Textbox(
+            lines=1,
+            label='FileName',
+            interactive=True,
+            show_copy_button=True,
+            )
+        delete_button = gr.Button(
+            variant='primary',
+            value='Delete',
+            )
+
+    with gr.Tab('embedding'):
+        gr.Markdown('データベースを更新します。')
+        embedding_button = gr.Button(
+            variant='primary',
+            value='Embedding',
+            )
+        gr.Markdown('メモリのデータベースを再読み込みします。')
+        reload_button = gr.Button(
+            variant='primary',
+            value='Reload',
+            )
 
     with gr.Tab('search'):
-        with gr.Row():
-            name = gr.Textbox(
-                lines=1,
-                label='name',
-                interactive=True,
-                show_copy_button=True,
-                )
-            query = gr.Textbox(
-                lines=1,
-                label='query',
-                interactive=True,
-                show_copy_button=True,
-                )
-            result = gr.Textbox(
-                label='result',
-                lines=20,
-                show_copy_button=True,
-                )
-        search_button = gr.Button(value='search')
+        search_name = gr.Textbox(
+            lines=1,
+            label='Name',
+            interactive=True,
+            show_copy_button=True,
+            )
+        search_query = gr.Textbox(
+            lines=1,
+            label='Query',
+            interactive=True,
+            show_copy_button=True,
+            )
+        search_button = gr.Button(
+            variant='primary',
+            value='Search',
+            )
+        search_result = gr.Textbox(
+            label='Result',
+            lines=20,
+            interactive=True,
+            show_copy_button=True,
+            )
+
+    ddg_button.click(
+        fn=fn.ddg,
+        inputs=[ddg_query],
+        outputs=[ddg_result],
+        )
+
+    bs4_button.click(
+        fn=fn.bs4,
+        inputs=[bs4_url],
+        outputs=[bs4_result],
+        )
+
+    upload_button.click(
+        fn=fn.upload,
+        inputs=[upload_name, upload_filename, upload_content],
+        outputs=[info],
+        )
+
+    delete_button.click(
+        fn=fn.delete,
+        inputs=[delete_name, delete_filename],
+        outputs=[info],
+        )
+
+    embedding_button.click(
+        fn=fn.embedding,
+        inputs=[],
+        outputs=[info],
+        )
+
+    reload_button.click(
+        fn=fn.load_vectors,
+        inputs=[],
+        outputs=[info],
+        )
 
     search_button.click(
         fn=fn.search,
-        inputs=[name, query],
-        outputs=[result],
+        inputs=[search_name, search_query],
+        outputs=[search_result],
         )
 
 if __name__ == '__main__':