feat: plotly map

app.py

@@ -2,41 +2,74 @@ 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')
-
-# Initialize Mistral client with the API key
 client = MistralClient(api_key=api_key)
+model = 'mistral-small'
+
 
-def answer_question(question):
-    """Directly ask Mistral the question and return the answer."""
-    # Format the user's question for Mistral
-    user_message = question
+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 ?"]
 
-    # Use the run_mistral function to get an answer
-    answer = run_mistral(user_message)
-    
-    return answer
 
-def run_mistral(user_message, model="mistral-medium"):
-    """Interact with Mistral using chat."""
-    messages = [ChatMessage(role="user", content=user_message)]
+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
 
-app = gr.Interface(
-    fn=chat_with_mistral,
-    inputs=gr.Textbox(lines=2, placeholder=placeholder),
-    outputs="text",
-    title=title,
-    description=description,
-    examples=examples
-)
+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")
 
-# Create a map using the library folium
+        # 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__":
-    app.launch(share=True)  # Set `share=True` to create a public link
+    demo.launch(share=True)  # Set `share=True` to create a public link

requirements.txt

@@ -2,4 +2,5 @@ python-dotenv
 mistralai
 langchain-community
 faiss-cpu
-gradio
+gradio
+plotly