Spaces:
Sleeping
Sleeping
| import os | |
| # Streamlit link: https://streamlit.io/ | |
| # Streamlit Docs link: https://docs.streamlit.io/ | |
| import streamlit as st | |
| from openai import OpenAI | |
| # Adding dotenv package: https://pypi.org/project/python-dotenv/ | |
| # pip install python-dotenv | |
| from dotenv import load_dotenv | |
| # Load environment variables | |
| load_dotenv() | |
| # Set the OpenAI API key from .env file | |
| client = OpenAI(api_key=os.getenv("OPENAI_API_KEY")) | |
| # Settings for the OpenAI model | |
| settings = { | |
| "model": "gpt-4o", | |
| "temperature": 0, | |
| "max_tokens": 100, | |
| "top_p": 1, | |
| "frequency_penalty": 0, | |
| "presence_penalty": 0 | |
| } | |
| def get_openai_response(user_message): | |
| response = client.chat.completions.create( | |
| model=settings["model"], | |
| messages=[ | |
| { | |
| "role": "system", | |
| # "content": "You are a helpful assistant in real-estate." | |
| "content": "You are a helpful bot, you always provide programming solutions in python in German, English or Spanish", | |
| }, | |
| { | |
| "role": "user", | |
| "content": user_message, | |
| }, | |
| ], | |
| temperature=settings["temperature"]) | |
| return response.choices[0].message.content | |
| # Streamlit App | |
| st.title("Chat with TechMaloo ChatBot") # Defining the app title in streamlit | |
| user_message = st.text_input("Enter your message:") | |
| if st.button("Send"): | |
| if user_message: | |
| with st.spinner("Generating response..."): | |
| try: | |
| response = get_openai_response(user_message) | |
| st.write(response) | |
| except Exception as e: | |
| st.write(f"Error: {e}") | |
| else: | |
| st.write("Please enter a message.") | |