Initial commit

README.md

@@ -1,5 +1,5 @@
 ---
-title: Openai Assistant Wanderlust
+title: OpenAI Assistant Wanderlust
 emoji: 🏢
 colorFrom: blue
 colorTo: green
@@ -11,3 +11,4 @@ license: cc-by-4.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+

app.py

@@ -0,0 +1,271 @@
+# -*- coding: utf-8 -*-
+"""OpenAI_Assistant_WanderLust.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1DLYjk07uQwF_Oqav1NtqMMCz_S2f0q8B
+
+# OpenAI assistant Wanderlust
+
+Basic recreation of OpenAI's DevDay Wanderlust demo app. It relies on Gradio and the new Assistants API.
+
+This space is inspired by the implementation of Fanilo Andrianasolo using Streamlit - https://www.youtube.com/watch?v=tLeqCDKgEDU
+"""
+
+#!pip install -q -U gradio openai datasets
+
+import gradio as gr
+import random
+import openai
+import os
+import json
+import plotly.graph_objects as go
+import time
+
+#os.environ["OPENAI_API_KEY"] = "..."  # Replace with your key
+
+#######################################
+# TOOLS SETUP
+#######################################
+
+def update_map_state(latitude, longitude, zoom):
+    """OpenAI tool to update map in-app
+    """
+    session_state[map_state] = {
+        "latitude": latitude,
+        "longitude": longitude,
+        "zoom": zoom,
+    }
+    print(session_state[map_state])
+    return "Map updated"
+
+def add_markers_state(latitudes, longitudes, labels):
+    """OpenAI tool to update markers in-app
+    """
+    session_state[markers_state] = {
+        "lat": latitudes,
+        "lon": longitudes,
+        "text": labels,
+    }
+    return "Markers added"
+
+tool_to_function = {
+    "update_map": update_map_state,
+    "add_markers": add_markers_state,
+}
+
+## Helpers
+
+def get_assistant_id():
+    return session_state[assistant_state].id
+
+def get_thread_id():
+    return session_state[thread_state].id
+
+
+def get_run_id():
+    return session_state[last_openai_run_state].id
+
+def submit_message(assistant_id, thread_id, user_message):
+    client.beta.threads.messages.create(
+        thread_id=thread_id, role="user", content=user_message
+    )
+    run = client.beta.threads.runs.create(
+        thread_id=thread_id,
+        assistant_id=assistant_id,
+    )
+    return run
+
+def get_run_info(run_id, thread_id):
+    run = client.beta.threads.runs.retrieve(
+        thread_id=thread_id,
+        run_id=run_id,
+    )
+    return run
+
+#######################################
+# SESSION SETUP
+#######################################
+
+client = openai.OpenAI()
+assistant_id = "asst_7OC3NTeyCjEZrApdLRklplE7"
+
+session_state = {}
+
+assistant_state = "assistant"
+thread_state = "thread"
+conversation_state = "conversation"
+last_openai_run_state = "last_openai_run"
+map_state = "map"
+markers_state = "markers"
+
+if (assistant_state not in session_state) or (thread_state not in session_state):
+    session_state[assistant_state] = client.beta.assistants.retrieve(assistant_id)
+    session_state[thread_state] = client.beta.threads.create()
+
+if conversation_state not in session_state:
+    session_state[conversation_state] = []
+
+if last_openai_run_state not in session_state:
+    session_state[last_openai_run_state] = None
+
+if map_state not in session_state:
+    session_state[map_state] = {
+        "latitude": 48.85,
+        "longitude": 2.35,
+        "zoom": 12,
+    }
+
+if markers_state not in session_state:
+    session_state[markers_state] = {
+        "lat": [],
+        "lon": [],
+        "text": [],
+    }
+
+fig = go.Figure(go.Scattermapbox())
+
+fig.update_layout(
+        mapbox_style="open-street-map",
+        hovermode='closest',
+        mapbox=dict(
+            center=go.layout.mapbox.Center(
+                lat=session_state[map_state]["latitude"],
+                lon=session_state[map_state]["longitude"]
+            ),
+            zoom=session_state[map_state]["zoom"]
+        ),
+)
+
+def respond(message, chat_history):
+
+    print(chat_history)
+
+    run = submit_message(get_assistant_id(), get_thread_id(), message)
+
+    session_state[last_openai_run_state] = run
+
+    print(run)
+
+    completed = False
+
+    # Polling
+    while not completed:
+
+        run = get_run_info(get_run_id(), get_thread_id())
+
+        if run.status == "requires_action":
+
+            tools_output = []
+
+            for tool_call in run.required_action.submit_tool_outputs.tool_calls:
+
+                f = tool_call.function
+                f_name = f.name
+                f_args = json.loads(f.arguments)
+
+                print(f"Launching function {f_name} with args {f_args}")
+
+                tool_result = tool_to_function[f_name](**f_args)
+
+                tools_output.append(
+                        {
+                            "tool_call_id": tool_call.id,
+                            "output": tool_result,
+                        }
+                    )
+
+            print(f"Will submit {tools_output}")
+
+            client.beta.threads.runs.submit_tool_outputs(
+                    thread_id=get_thread_id(),
+                    run_id=get_run_id(),
+                    tool_outputs=tools_output,
+            )
+
+        if run.status == "completed":
+
+            completed = True
+
+        else:
+            time.sleep(0.1)
+
+    session_state[conversation_state] = [
+        [m.role, m.content[0].text.value]
+        for m in client.beta.threads.messages.list(get_thread_id(), order="asc").data
+    ]
+
+    dialog = session_state[conversation_state]
+    formatted_dialog = []
+    for i in range(int(len(dialog)/2)):
+      formatted_dialog.append([dialog[i*2][1],dialog[i*2+1][1]])
+
+
+    chat_history = formatted_dialog
+
+    fig = None
+
+    if session_state[markers_state] is None:
+
+      fig = go.Figure(go.Scattermapbox())
+
+    else :
+      fig = go.Figure(go.Scattermapbox(
+            customdata=session_state[markers_state]["text"],
+            lat=session_state[markers_state]["lat"],
+            lon=session_state[markers_state]["lon"],
+            mode='markers',
+            marker=go.scattermapbox.Marker(
+                size=18
+            ),
+            hoverinfo="text",
+            hovertemplate='<b>Name</b>: %{customdata}'
+        ))
+
+    fig.update_layout(
+        mapbox_style="open-street-map",
+        hovermode='closest',
+        mapbox=dict(
+            center=go.layout.mapbox.Center(
+                lat=session_state[map_state]["latitude"],
+                lon=session_state[map_state]["longitude"]
+            ),
+            zoom=12
+        ),
+    )
+
+    return "", chat_history, fig
+
+with gr.Blocks(title="OpenAI assistant Wanderlust") as demo:
+
+    gr.Markdown("# OpenAI assistant Wanderlust")
+
+    with gr.Column():
+        with gr.Row():
+
+          chatbot = gr.Chatbot()
+          map = gr.Plot(fig)
+
+    msg = gr.Textbox("Move the map to Brussels and add markers for five major attractions")
+
+    with gr.Column():
+        with gr.Row():
+          submit = gr.Button("Submit")
+          clear = gr.ClearButton([msg, chatbot])
+
+    msg.submit(respond, [msg, chatbot], [msg, chatbot, map])
+    submit.click(respond, [msg, chatbot], [msg, chatbot, map])
+
+    gr.Markdown(
+        """
+# Description
+
+Basic recreation of OpenAI's DevDay Wanderlust demo app. It relies on Gradio and the new Assistants API. [Github repository](https://github.com/Yannael/openai-assistant-wanderlust)
+
+This space is inspired by the implementation of [Fanilo Andrianasolo using Streamlit](https://www.youtube.com/watch?v=tLeqCDKgEDU)
+"""
+    )
+
+demo.launch(debug=True)
+

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+openai
+datasets
+plotly