Update app.py

app.py

@@ -1,5 +1,31 @@
 import gradio as gr
-from src import requests
+from mistralai.client import MistralClient
+from mistralai.models.chat_completion import ChatMessage
+
+
+def get_stream_chat_completion(
+    message, chat_history, model, api_key, system=None, **kwargs
+):
+    messages = []
+    if system is not None:
+        messages.append(ChatMessage(role="system", content=system))
+    for chat in chat_history:
+        human_message, bot_message = chat
+        messages.extend(
+            (
+                ChatMessage(role="user", content=human_message),
+                ChatMessage(role="assistant", content=bot_message),
+            )
+        )
+    messages.append(ChatMessage(role="user", content=message))
+    client = MistralClient(api_key=api_key)
+    for chunk in client.chat_stream(
+        model=model,
+        messages=messages,
+        **kwargs,
+    ):
+        if chunk.choices[0].delta.content is not None:
+            yield chunk.choices[0].delta.content
 
 def respond_stream(
     message,
@@ -13,7 +39,7 @@ def respond_stream(
 ):
     response = ""
     received_anything = False
-    for chunk in requests.get_stream_chat_completion(
+    for chunk in get_stream_chat_completion(
         message=message,
         chat_history=chat_history,
         model=model,