ATAfrica / app.py
Stanlito's picture
Upload 3 files
698a47e verified
raw
history blame contribute delete
No virus
1.45 kB
import os
from dotenv import load_dotenv
import streamlit as st
import openai
# Function to chat with GPT
def chat_with_gpt(prompt):
load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")
result = openai.chat.completions.create(
model="gpt-3.5-turbo",
messages=[
{
"role": "user",
"content": prompt
}
]
)
return result.choices[0].message.content
# Streamlit UI
def main():
st.title("Generation and Analysis Tool with OpenAI")
# Add text input field
prompt = st.text_input("Please enter a question or request:")
# Interaction options
max_tokens = st.sidebar.slider("Max Tokens", min_value=10, max_value=200, value=50, step=10,
help="Maximum number of tokens to generate")
temperature = st.sidebar.slider("Temperature", min_value=0.1, max_value=1.0, value=0.5, step=0.1,
help="Controls the randomness of the generated text. Higher values make the text "
"more random.")
# Add a button to trigger the chat
if st.button("Enter"):
if prompt.strip() == '':
st.warning("Please enter a valid prompt.")
else:
response = chat_with_gpt(prompt)
st.success("Here's the response:")
st.write(response)
if __name__ == "__main__":
main()