dzivkovi commited on
Commit
2c3f7a8
1 Parent(s): 641c708

Added code from Kristopher Overholt

Browse files
Files changed (5) hide show
  1. .gitignore +162 -0
  2. Procfile +1 -0
  3. README.md +55 -1
  4. app.py +52 -4
  5. images/chatbot.png +0 -0
.gitignore ADDED
@@ -0,0 +1,162 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Byte-compiled / optimized / DLL files
2
+ __pycache__/
3
+ *.py[cod]
4
+ *$py.class
5
+
6
+ # C extensions
7
+ *.so
8
+
9
+ # Distribution / packaging
10
+ .Python
11
+ build/
12
+ develop-eggs/
13
+ dist/
14
+ downloads/
15
+ eggs/
16
+ .eggs/
17
+ lib/
18
+ lib64/
19
+ parts/
20
+ sdist/
21
+ var/
22
+ wheels/
23
+ share/python-wheels/
24
+ *.egg-info/
25
+ .installed.cfg
26
+ *.egg
27
+ MANIFEST
28
+
29
+ # PyInstaller
30
+ # Usually these files are written by a python script from a template
31
+ # before PyInstaller builds the exe, so as to inject date/other infos into it.
32
+ *.manifest
33
+ *.spec
34
+
35
+ # Installer logs
36
+ pip-log.txt
37
+ pip-delete-this-directory.txt
38
+
39
+ # Unit test / coverage reports
40
+ htmlcov/
41
+ .tox/
42
+ .nox/
43
+ .coverage
44
+ .coverage.*
45
+ .cache
46
+ nosetests.xml
47
+ coverage.xml
48
+ *.cover
49
+ *.py,cover
50
+ .hypothesis/
51
+ .pytest_cache/
52
+ cover/
53
+
54
+ # Translations
55
+ *.mo
56
+ *.pot
57
+
58
+ # Django stuff:
59
+ *.log
60
+ local_settings.py
61
+ db.sqlite3
62
+ db.sqlite3-journal
63
+
64
+ # Flask stuff:
65
+ instance/
66
+ .webassets-cache
67
+
68
+ # Scrapy stuff:
69
+ .scrapy
70
+
71
+ # Sphinx documentation
72
+ docs/_build/
73
+
74
+ # PyBuilder
75
+ .pybuilder/
76
+ target/
77
+
78
+ # Jupyter Notebook
79
+ .ipynb_checkpoints
80
+
81
+ # IPython
82
+ profile_default/
83
+ ipython_config.py
84
+
85
+ # pyenv
86
+ # For a library or package, you might want to ignore these files since the code is
87
+ # intended to run in multiple environments; otherwise, check them in:
88
+ # .python-version
89
+
90
+ # pipenv
91
+ # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
92
+ # However, in case of collaboration, if having platform-specific dependencies or dependencies
93
+ # having no cross-platform support, pipenv may install dependencies that don't work, or not
94
+ # install all needed dependencies.
95
+ #Pipfile.lock
96
+
97
+ # poetry
98
+ # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
99
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
100
+ # commonly ignored for libraries.
101
+ # https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
102
+ #poetry.lock
103
+
104
+ # pdm
105
+ # Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
106
+ #pdm.lock
107
+ # pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
108
+ # in version control.
109
+ # https://pdm.fming.dev/#use-with-ide
110
+ .pdm.toml
111
+
112
+ # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
113
+ __pypackages__/
114
+
115
+ # Celery stuff
116
+ celerybeat-schedule
117
+ celerybeat.pid
118
+
119
+ # SageMath parsed files
120
+ *.sage.py
121
+
122
+ # Environments
123
+ .env
124
+ .venv
125
+ env/
126
+ venv/
127
+ ENV/
128
+ env.bak/
129
+ venv.bak/
130
+
131
+ # Spyder project settings
132
+ .spyderproject
133
+ .spyproject
134
+
135
+ # Rope project settings
136
+ .ropeproject
137
+
138
+ # mkdocs documentation
139
+ /site
140
+
141
+ # mypy
142
+ .mypy_cache/
143
+ .dmypy.json
144
+ dmypy.json
145
+
146
+ # Pyre type checker
147
+ .pyre/
148
+
149
+ # pytype static type analyzer
150
+ .pytype/
151
+
152
+ # Cython debug symbols
153
+ cython_debug/
154
+
155
+ # PyCharm
156
+ # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
157
+ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
158
+ # and can be added to the global gitignore or merged into this file. For a more nuclear
159
+ # option (not recommended) you can uncomment the following to ignore the entire idea folder.
160
+ #.idea/
161
+
162
+ .DS_Store
Procfile ADDED
@@ -0,0 +1 @@
 
 
1
+ web: python app.py
README.md CHANGED
@@ -9,5 +9,59 @@ app_file: app.py
9
  pinned: false
10
  license: mit
11
  ---
 
12
 
13
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  pinned: false
10
  license: mit
11
  ---
12
+ # palm-vertex-ai-chatbot
13
 
14
+ An example chatbot app with Gradio that uses the PaLM API from Vertex AI in
15
+ Google Cloud
16
+
17
+ ![Chatbot app powered by the PaLM API in Google Cloud](images/chatbot.png)
18
+
19
+ ## Usage
20
+
21
+ 1. Clone this repository.
22
+
23
+ 2. Install the required dependencies:
24
+
25
+ ```bash
26
+ pip install -r requirements.txt
27
+ ```
28
+
29
+ 3. In the GCP Console, navigate to `Artificial Intelligence > Vertex AI > Dashboard`, and enable the
30
+ [Vertex AI API](https://console.cloud.google.com/apis/enableflow?apiid=aiplatform.googleapis.com).
31
+
32
+ 4. Optionally, if you want to run the Streamlit app on Railway, navigate to `APIs & Services > Enabled APIs & services`, and select Vertex AI API from the list. From the `Credentials tab`, click `Create Credentials > Service account`
33
+ (https://console.cloud.google.com/iam-admin/serviceaccounts/create)
34
+ , and create a service account with the Vertex AI User role. Finally, add and download a new key for this service account i.e. JSON credentials file.
35
+
36
+ 5. Set the `GOOGLE_APPLICATION_CREDENTIALS` environment variable, e.g.
37
+
38
+ ```bash
39
+ export GOOGLE_APPLICATION_CREDENTIALS='/tmp/vertex-ai-user.json'
40
+ ```
41
+
42
+ 4. Run the app:
43
+
44
+ ```bash
45
+ python app.py
46
+ ```
47
+
48
+ 5. Navigate to `http://127.0.0.1:8080/` in your browser.
49
+
50
+ 6. Start chatting with your PaLM-powered chatbot!
51
+
52
+ ## Deploy to Cloud Run
53
+
54
+ The above steps allow you to test and use the chatbot app on your local machine.
55
+ You can use a fully managed service such as Cloud Run to publish your app.
56
+
57
+ 1. Run the following command using the
58
+ [gcloud CLI](https://cloud.google.com/sdk/gcloud):
59
+
60
+ ```bash
61
+ gcloud run deploy chatbot-app --source . --allow-unauthenticated --region us-central1
62
+ ```
63
+
64
+ 2. After a couple of minutes, your chatbot app will be deployed on Cloud Run,
65
+ and you can access the app via a URL similar to:
66
+
67
+ [`https://chatbot-app-r5gdynozbq-uc.a.run.app/`](https://chatbot-app-r5gdynozbq-uc.a.run.app/)
app.py CHANGED
@@ -1,7 +1,55 @@
1
  import gradio as gr
 
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- iface.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ from vertexai.preview.language_models import ChatModel, InputOutputTextPair, ChatSession
3
 
 
 
4
 
5
+ def create_session():
6
+ chat_model = ChatModel.from_pretrained("chat-bison@001")
7
+ parameters = {
8
+ "temperature": 0.2,
9
+ "max_output_tokens": 256,
10
+ "top_k": 40,
11
+ "top_p": 0.80,
12
+ "context": "You are an astronomer, knowledgeable about the solar system.",
13
+ "examples": [
14
+ InputOutputTextPair(
15
+ input_text="How many moons does Mars have?",
16
+ output_text="The planet Mars has two moons, Phobos and Deimos.",
17
+ ),
18
+ ],
19
+ }
20
+ chat = ChatSession(model=chat_model, **parameters)
21
+ return chat
22
+
23
+
24
+ def response(chat, message):
25
+ res = chat.send_message(
26
+ message=message, max_output_tokens=256, temperature=0.2, top_k=40, top_p=0.8
27
+ )
28
+ return res.text
29
+
30
+
31
+ with gr.Blocks() as demo:
32
+ chatbot = gr.Chatbot(
33
+ [
34
+ (
35
+ "Hello! I'm an astronomer chatbot.\nAsk me anything about the solar system!",
36
+ None,
37
+ )
38
+ ],
39
+ label="PaLM-based Chatbot powered by Google Cloud",
40
+ ).style(height=400)
41
+ msg = gr.Textbox(placeholder="Type your response here", label="Response")
42
+ clear = gr.Button("Clear")
43
+ chat_model = create_session()
44
+
45
+ def respond(message, chat_history):
46
+ texts = [message]
47
+ bot_message = response(chat_model, message)
48
+ chat_history.append((message, bot_message))
49
+ return "", chat_history
50
+
51
+ msg.submit(respond, [msg, chatbot], [msg, chatbot])
52
+ clear.click(lambda: None, None, chatbot, queue=False)
53
+
54
+ if __name__ == "__main__":
55
+ demo.launch(server_port=8080)
images/chatbot.png ADDED