Spaces:
Paused
Paused
File size: 2,267 Bytes
e107fb2 8f27bd5 aeaf613 8f27bd5 aeaf613 8f27bd5 aeaf613 f8f1f43 8f27bd5 aeaf613 8f27bd5 e107fb2 aeaf613 754727f aeaf613 754727f e107fb2 aeaf613 |
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 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
import gradio as gr
from mistralai.client import MistralClient, ChatMessage
import os
from dotenv import load_dotenv
import plotly.graph_objects as go
# Load environment variables
load_dotenv()
api_key = os.getenv('API_KEY')
client = MistralClient(api_key=api_key)
model = 'mistral-small'
title = "Gaia Mistral Chat Demo"
description = "Example of simple chatbot with Gradio and Mistral AI via its API"
placeholder = "Posez moi une question sur l'agriculture"
examples = ["Comment fait on pour produire du maïs ?", "Rédige moi une lettre pour faire un stage dans une exploitation agricole", "Comment reprendre une exploitation agricole ?"]
def chat_with_mistral(user_input):
messages = [ChatMessage(role="user", content=user_input)]
chat_response = client.chat(model=model, messages=messages)
return chat_response.choices[0].message.content
def create_world_map(
lat=45.5017,
lon=-73.5673,
):
fig = go.Figure(go.Scattermapbox
(
lat=[lat],
lon=[lon],
mode='markers',
marker=go.scattermapbox.Marker(size=14),
text=['Montreal'],
))
fig.update_layout(
mapbox_style="open-street-map",
hovermode='closest',
mapbox=dict(
bearing=0,
center=go.layout.mapbox.Center(
lat=lat,
lon=lon,
),
pitch=0,
zoom=5
),)
return fig
with gr.Blocks() as demo:
with gr.Column():
with gr.Row():
user_input = gr.Textbox(lines=2, placeholder=placeholder)
send_chat_btn = gr.Button(value="Send")
lat = gr.Number(value=45.5017, label="Latitude")
lon = gr.Number(value=-73.5673, label="Longitude")
update_map_btn = gr.Button(value="Update Map")
chat_output = gr.Textbox(lines=2, placeholder="Réponse")
# map:
map = gr.Plot()
demo.load(chat_with_mistral, user_input, chat_output)
send_chat_btn.click(chat_with_mistral, user_input, chat_output)
# map:
demo.load(create_world_map, [lat, lon], map)
update_map_btn.click(create_world_map, [lat, lon], map)
if __name__ == "__main__":
demo.launch(share=True) # Set `share=True` to create a public link
|