Upload 2 files

app.py

@@ -2,7 +2,8 @@ from langchain.agents import AgentExecutor, AgentType, initialize_agent
 from langchain.agents.structured_chat.prompt import SUFFIX
 from langchain.chat_models import ChatOpenAI
 from langchain.memory import ConversationBufferMemory
-from tools import generate_image_tool, describe_image_tool, handle_image_history, wait_for_key
+from tools import generate_image_tool, describe_image_tool, handle_image_history
+from openai import OpenAI
 
 import chainlit as cl
 from chainlit.action import Action
@@ -18,6 +19,30 @@ def get_memory():
     return ConversationBufferMemory(memory_key="chat_history")
 
 
+async def wait_for_key():
+    res = await cl.AskUserMessage(content="Send an Openai API KEY to start. [https://platform.openai.com/api-keys](https://platform.openai.com/api-keys). e.g. sk-IY8Wl.....1cXD8", timeout=600).send()
+    if res:
+        await cl.Message(content="setting up...", indent=1).send()
+        # check if the key is valid
+        client = OpenAI(api_key=res["content"])
+        try:
+            stream = client.chat.completions.create(
+                model="gpt-3.5-turbo-1106",
+                messages=[{"role": "system", "content": "test"}],
+                max_tokens=1,
+            )
+            if stream:
+                await cl.Message(content="API_KEY setted, you can start chatting!", indent=1).send()
+                cl.user_session.set("api_key", res["content"])
+        except Exception as e:
+            await cl.Message(content=f"{e}", indent=1).send()
+            return await wait_for_key()
+        await cl.Message(content="✅ API KEY works! ✅ you can start chatting! 💬").send()
+        return res["content"]
+    else:
+        return await wait_for_key()
+
+
 @cl.on_chat_start
 async def start():
     """
@@ -26,7 +51,7 @@ async def start():
     We can add some settings to our application to allow users to select the appropriate model, and more!
     """
     cl.user_session.set("image_history", [{"role": "system", "content": "You are a helpful assistant. You are developed with GPT-4-vision-preview, if the user uploads an image, you have the ability to understand it."}])
-    cl.user_session.set("api_key", "sk-xxx")
+
     settings = await cl.ChatSettings(
         [
             Select(
@@ -46,8 +71,9 @@ async def start():
             ),
         ]
     ).send()
-    await setup_agent(settings)
-    await wait_for_key()
+    api_key = await wait_for_key()
+    if api_key:
+        await setup_agent(settings)
 
 
 @cl.on_settings_update

tools.py

@@ -130,27 +130,3 @@ describe_image_tool = Tool.from_function(
     description=f"Useful to describe an image. Input should be a single string strictly in the following JSON format: {describe_image_format}",
     return_direct=False,
 )
-
-
-
-async def wait_for_key():
-    res = await cl.AskUserMessage(content="Send an Openai API KEY to start. [https://platform.openai.com/api-keys](https://platform.openai.com/api-keys). e.g. sk-IY8Wl.....1cXD8", timeout=600).send()
-    if res:
-        await cl.Message(content="setting up...", indent=1).send()
-        # check if the key is valid
-        client = OpenAI(api_key=res["content"])
-        try:
-            stream = client.chat.completions.create(
-                model="gpt-3.5-turbo-1106",
-                messages=[{"role": "system", "content": "test"}],
-                max_tokens=1,
-            )
-            if stream:
-                await cl.Message(content="API_KEY setted, you can start chatting!", indent=1).send()
-                cl.user_session.set("api_key", res["content"])
-        except Exception as e:
-            await cl.Message(content=f"{e}", indent=1).send()
-            return await wait_for_key()
-        return await cl.Message(content="✅ API KEY works! ✅ you can start chatting! 💬").send()
-    else:
-        return await wait_for_key()