import streamlit as st
import openai

# Set your OpenAI API key
openai.api_key = 'sk-proj-Jd8bv8afFkjunDur6HD9T3BlbkFJ5s9heeKGsQxC5wMEbgcI'

st.title("Simple Chatbot with Fine-Tuned OpenAI Model")

if 'history' not in st.session_state:
    st.session_state.history = []

def generate_response(prompt):
    response = openai.Completion.create(
        model="fine-tuned-model-id",  # Use your fine-tuned model ID
        prompt=prompt,
        max_tokens=150
    )
    return response.choices[0].text.strip()

user_input = st.text_input("You:", key='input')

if user_input:
    st.session_state.history.append({"role": "user", "content": user_input})
    response = generate_response(user_input)
    st.session_state.history.append({"role": "bot", "content": response})

for message in st.session_state.history:
    if message['role'] == 'user':
        st.write(f"You: {message['content']}")
    else:
        st.write(f"Bot: {message['content']}")