Tanu007 commited on
Commit
51ca969
·
verified ·
1 Parent(s): ff60b23

Upload 2 files

Browse files
Files changed (2) hide show
  1. .env +1 -0
  2. main.py +40 -0
.env ADDED
@@ -0,0 +1 @@
 
 
1
+ MISTRAL_API = "5Tr1VHkHgUvXDDT1yH5N1O7kbhIiwJuc"
main.py ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import gradio as gr
3
+ from mistralai import Mistral, UserMessage
4
+ from dotenv import load_dotenv, find_dotenv
5
+
6
+
7
+ def chat_bot(user_input):
8
+ load_dotenv(find_dotenv())
9
+ miss_api_key = os.environ["MISTRAL_API"]
10
+ model = "mistral-small-latest"
11
+
12
+ client = Mistral(api_key=miss_api_key)
13
+
14
+ message = [{
15
+ "role": "user",
16
+ "content": user_input,
17
+ }, ]
18
+
19
+ chat_response = client.chat.complete(
20
+ model=model,
21
+ messages=message,
22
+ )
23
+ # response = client.create_chat_completion()
24
+ return chat_response.choices[0].message.content
25
+
26
+ # for chunk in client.chat.stream(model=model, messages=message):
27
+ # if "delta" in chunk.data.choices[0]:
28
+ # if chunk.data.choices[0].delta.content:
29
+ # print(chunk.data.choices[0].delta.content, end="")
30
+
31
+
32
+ iface = gr.Interface(
33
+ fn=chat_bot,
34
+ inputs="text",
35
+ outputs="text",
36
+ description="Guides the website usage",
37
+ examples=[["How can you help me?"]],
38
+ title="Website Guide"
39
+ )
40
+ iface.launch()