vince1313 commited on
Commit
f0bc472
·
verified ·
1 Parent(s): c40e0b7

My first commit

Browse files
Files changed (1) hide show
  1. app.py +61 -0
app.py ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ # Streamlit link: https://streamlit.io/
4
+ # Streamlit Docs link: https://docs.streamlit.io/
5
+ import streamlit as st
6
+ from openai import OpenAI
7
+
8
+ # Adding dotenv package: https://pypi.org/project/python-dotenv/
9
+ # pip install python-dotenv
10
+ from dotenv import load_dotenv
11
+
12
+ # Load environment variables
13
+ load_dotenv()
14
+
15
+ # Set the OpenAI API key from .env file
16
+ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
17
+
18
+ # Settings for the OpenAI model
19
+ settings = {
20
+ "model": "gpt-4o",
21
+ "temperature": 0,
22
+ "max_tokens": 100,
23
+ "top_p": 1,
24
+ "frequency_penalty": 0,
25
+ "presence_penalty": 0
26
+ }
27
+
28
+
29
+ def get_openai_response(user_message):
30
+ response = client.chat.completions.create(
31
+ model=settings["model"],
32
+ messages=[
33
+ {
34
+ "role": "system",
35
+ # "content": "You are a helpful assistant in real-estate."
36
+ "content": "You are a helpful bot, you always provide programming solutions in python in German, English or Spanish",
37
+ },
38
+ {
39
+ "role": "user",
40
+ "content": user_message,
41
+ },
42
+ ],
43
+ temperature=settings["temperature"])
44
+ return response.choices[0].message.content
45
+
46
+
47
+ # Streamlit App
48
+ st.title("Chat with TechMalooBot") # Defining the app title in streamlit
49
+ user_message = st.text_input("Enter your message:")
50
+ if st.button("Send"):
51
+ if user_message:
52
+ with st.spinner("Generating response..."):
53
+ try:
54
+ response = get_openai_response(user_message)
55
+ st.write(response)
56
+ except Exception as e:
57
+ st.write(f"Error: {e}")
58
+ else:
59
+ st.write("Please enter a message.")
60
+
61
+