Update app.py

app.py

@@ -1,21 +1,16 @@
 """
 Simple Chatbot
-@author: Nigel Gebodh
-@email: [email protected]
 """
 
 import streamlit as st
-from openai import OpenAI
+import openai  # Ensure you have the correct import
 import os
 from dotenv import load_dotenv
 
 load_dotenv()
 
 # Initialize the client
-client = OpenAI(
-    base_url="https://api-inference.huggingface.co/v1",
-    api_key=os.environ.get('HUGGINGFACEHUB_API_TOKEN')  # Replace with your token
-)
+openai.api_key = os.environ.get('HUGGINGFACEHUB_API_TOKEN')  # Replace with your token
 
 model_link = "mistralai/Mistral-7B-Instruct-v0.2"
 
@@ -57,18 +52,16 @@ if prompt:
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
         try:
-            response = client.chat.completions.create(
-                model=model_link,
-                messages=[
-                    {"role": m["role"], "content": m["content"]}
-                    for m in st.session_state.messages
-                ],
-                temperature=temperature,
-                max_tokens=3000
-            )['choices'][0]['message']['content']
+            response = openai.Completion.create(
+                engine=model_link,
+                prompt=prompt,
+                max_tokens=3000,
+                temperature=temperature
+            )
 
-            st.markdown(response)
-            st.session_state.messages.append({"role": "assistant", "content": response})
+            response_content = response.choices[0].text.strip()
+            st.markdown(response_content)
+            st.session_state.messages.append({"role": "assistant", "content": response_content})
 
         except Exception as e:
             st.markdown("An error occurred. Please try again later.")