init

.gitjgnore

@@ -0,0 +1,141 @@
+.vscode
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+

.vscode/settings.json

@@ -0,0 +1,9 @@
+{
+    "workbench.colorCustomizations": {
+        "editor.lineHighlightBackground": "#1073cf2d",
+        "editor.lineHighlightBorder": "#9fced11f",
+        "activityBar.background": "#312E17",
+        "titleBar.activeBackground": "#454120",
+        "titleBar.activeForeground": "#FBFAF6"
+    }
+}

app.py

@@ -0,0 +1,59 @@
+# !/usr/bin/env python
+# -*- coding:utf-8 -*-
+# ==================================================================
+# [CreatedDate]  : Thursday, 1970-01-01 08:00:00
+# [Author]       : shixiaofeng
+# [Descriptions] :
+# ==================================================================
+# [ChangeLog]:
+# [Date]    	[Author]	[Comments]
+# ------------------------------------------------------------------
+
+
+import json
+import logging
+import time
+
+import gradio as gr
+import requests
+
+logger = logging.getLogger('gradio')
+gr.close_all()
+
+host = "127.0.0.1:9172"
+
+
+def tableqa(input_question, history=""):
+    logger.info("run tableqa")
+    if history == "":
+        history_sql = None
+    else:
+        history_sql = json.loads(history.replace("\'", "\""))
+    data = {"raw_data": {'question': input_question, 'history_sql': history_sql}}
+
+    ts = time.time()
+    r = requests.post(f"http://{host}/tableqa", json=data)
+    response = json.loads(r.text)
+    print("response", response)
+    te = time.time()
+    print("run inference_mask_sam success [{}], time_Cost is [{}]".format(response["code"] == 200, te-ts))
+
+    if response["code"] == 200:
+        df_value = response["result"]["select_df"]
+        df = {"data":df_value["rows"],"headers":df_value["header_name"]}
+        return [df, response["result"]["sql_string"], response["result"]["sql_query"], response["result"]["history"], response["result"]["query_result"]]
+    else:
+        return ["1", "2", "3", "4",]
+
+
+# iface = gr.Interface(fn=greet, inputs="text", outputs=["输出sql语句","输出可执行sql语句","执行结果"])
+iface = gr.Interface(fn=tableqa, inputs=[gr.Textbox(label="input_question", info="请输入想要查询的问题."),
+                                         gr.Textbox(label="history sql", info="上下文对话历史信息.")],
+                      outputs=[gr.DataFrame(label="索引到的数据库"),
+                               gr.Textbox(label="输出sql语句"),
+                               gr.Textbox(label="输出可执行sql语句"),
+                               gr.Textbox(label="多轮对话历史sql"),
+                               gr.Textbox(label="SQL执行结果")]
+                     )
+
+iface.launch(enable_queue=False, server_name="0.0.0.0", server_port=9176, debug=True)