File size: 1,664 Bytes
908d449 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
from typing import List, Tuple
import gradio as gr
import google.generativeai as genai
def predict(google_key: str, text_prompt: str, chatbot: List[Tuple[str, str]]):
if not google_key:
raise ValueError(
"GOOGLE_API_KEY is not set. "
"Please follow the instructions in the README to set it up.")
genai.configure(api_key=google_key)
model = genai.GenerativeModel('models/gemini-pro')
response = model.generate_content(text_prompt, stream=True)
response.resolve()
chatbot.append((text_prompt, response.text))
return "", chatbot
google_key_component = gr.Textbox(
label="GOOGLE API KEY",
value="",
type="password",
placeholder="...",
info="You have to provide your own GPT4 keys for this app to function properly",
)
chatbot_component = gr.Chatbot(label='Gemini')
text_prompt_component = gr.Textbox(
placeholder="Hi there!",
label="Type an input and press Enter"
)
run_button_component = gr.Button()
with gr.Blocks() as demo:
with gr.Column():
google_key_component.render()
with gr.Row():
chatbot_component.render()
text_prompt_component.render()
run_button_component.render()
run_button_component.click(
fn=predict,
inputs=[google_key_component, text_prompt_component, chatbot_component],
outputs=[text_prompt_component, chatbot_component],
)
text_prompt_component.submit(
fn=predict,
inputs=[google_key_component, text_prompt_component, chatbot_component],
outputs=[text_prompt_component, chatbot_component],
)
demo.queue(max_size=99).launch(debug=True)
|