Update app.py

app.py

@@ -5,8 +5,75 @@ import os
 from dotenv import load_dotenv
 load_dotenv()
 import openai
+from openai import ChatCompletion
 
 openai.api_key = os.getenv('OPENAI_API_KEY')
 # keys are here: https://platform.openai.com/auth/callback?code=ReZ4izEw0DwkUKrHR-Opxr5AMMgo9SojxC9pNHQUcjD6M&state=OGZFNDJmLlJGNlIwOUxlakpXZkVFfjNxNy02ZlFtLWN4eUcuOXJobXouSQ%3D%3D#
 # Read this, https://blog.futuresmart.ai/building-a-gpt-4-chatbot-using-chatgpt-api-and-streamlit-chat
 
+import streamlit as st
+
+# Set the API key (replace "YOUR_API_KEY" with your actual OpenAI API key)
+openai.api_key = 'YOUR_API_KEY'
+
+
+
+
+# Define a function to chat with the model
+def chat_with_model(prompts):
+    model = "gpt-3.5-turbo"  # change this to the model you're using
+
+    conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
+    conversation.extend([{'role': 'user', 'content': prompt} for prompt in prompts])
+
+    response = openai.ChatCompletion.create(model=model, messages=conversation)
+    return response['choices'][0]['message']['content']
+
+
+import openai
+import streamlit as st
+from streamlit_chat import message
+import os 
+from dotenv import load_dotenv
+load_dotenv('api_key.env')
+openai.api_key = os.environ.get('API_KEY')
+
+
+
+
+def generate_response(prompt):
+    completion=openai.Completion.create(
+        engine='text-davinci-003',
+        prompt=prompt,
+        max_tokens=1024,
+        n=1,
+        stop=None,
+        temperature=0.6,
+    )
+    message=completion.choices[0].text
+    return message
+
+
+    
+    
+# Streamlit App
+def main():
+    st.title("Chat with AI")
+
+    # Pre-defined prompts
+    prompts = ['How's the weather?', 'Tell me a joke.', 'What is the meaning of life?']
+
+    # User prompt input
+    user_prompt = st.text_input("Your question:", '')
+
+    if user_prompt:
+        prompts.append(user_prompt)
+
+    if st.button('Chat'):
+        st.write('Chatting with GPT-3...')
+        response = chat_with_model(prompts)
+        st.write('Response:')
+        st.write(response)
+
+if __name__ == "__main__":
+    main()