vince1313 commited on
Commit
9ccd396
·
verified ·
1 Parent(s): 92a90f3

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +51 -0
app.py ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import streamlit as st
3
+ from openai import OpenAI
4
+
5
+ from dotenv import load_dotenv
6
+
7
+ # Load environment variables
8
+ load_dotenv()
9
+
10
+ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
11
+
12
+ # Set the OpenAI API key
13
+
14
+ # Settings for the OpenAI model
15
+ settings = {
16
+ "model": "gpt-4",
17
+ "temperature": 0,
18
+ # ... more settings if needed
19
+ }
20
+
21
+
22
+ def get_openai_response(user_message):
23
+ response = client.chat.completions.create(model=settings["model"],
24
+ messages = [
25
+ {
26
+ "role": "system",
27
+ "content": "You are a helpful bot, you always provide programming solutions in python in English",
28
+ },
29
+ {
30
+ "role": "user",
31
+ "content": user_message,
32
+ },
33
+ ],
34
+ temperature=settings["temperature"])
35
+ return response.choices[0].message.content
36
+
37
+
38
+ st.title("Chat with TechMalooBot")
39
+ user_message = st.text_input("Enter your message:")
40
+ if st.button("Send"):
41
+ if user_message:
42
+ with st.spinner("Generating response..."):
43
+ try:
44
+ response = get_openai_response(user_message)
45
+ st.write(response)
46
+ except Exception as e:
47
+ st.write(f"Error: {e}")
48
+ else:
49
+ st.write("Please enter a message.")
50
+
51
+