| import os |
| import streamlit as st |
| from openai import OpenAI |
|
|
| from dotenv import load_dotenv |
|
|
| |
| load_dotenv() |
|
|
| client = OpenAI(api_key=os.getenv("OPENAI_API_KEY")) |
|
|
| |
|
|
| |
| settings = { |
| "model": "gpt-4", |
| "temperature": 0, |
| |
| } |
|
|
|
|
| def get_openai_response(user_message): |
| response = client.chat.completions.create(model=settings["model"], |
| messages = [ |
| { |
| "role": "system", |
| "content": "You are a helpful bot, you always provide programming solutions in python in English", |
| }, |
| { |
| "role": "user", |
| "content": user_message, |
| }, |
| ], |
| temperature=settings["temperature"]) |
| return response.choices[0].message.content |
|
|
|
|
| st.title("Chat with TechMalooBot") |
| 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.") |
|
|
