Update app.py

app.py

@@ -1,51 +1,44 @@
+`
 import streamlit as st
-from huggingface_hub import InferenceClient, InferenceTimeoutError
-import asyncio
+import Cohere
+from streamlit_chat import message
 
-# Initialize the Inference Client
-model_id = "gpt-3.5-turbo"  # Replace with your desired model ID
-api_token = "gym8F1vSoz4RazTSa8D9YjuINil8VLV6oUHqJ2jh"  # Replace with your actual Hugging Face API token
 
-client = InferenceClient(model=model_id, token=api_token)
+def api_calling(question):
+    model = Cohere(cohere_api_key='gym8F1vSoz4RazTSa8D9YjuINil8VLV6oUHqJ2jh')
+    response = model.invoke(question)
+    return  response  
 
+st.title("COHERE CHATBOT BADR")
+if 'user_input' not in st.session_state:
+	st.session_state['user_input'] = []
 
-# Function to get response synchronously
-def get_response_sync(prompt):
-    try:
-        response = client.text_generation(prompt=prompt)
-        return response[0]['generated_text']
-    except InferenceTimeoutError as e:
-        return f"Request timed out: {e}"
+if 'cohere_response' not in st.session_state:
+	st.session_state['cohere_response'] = []
 
-# Function to get response asynchronously
-async def get_response_async(prompt):
-    try:
-        response = await client.text_generation_async(prompt=prompt)
-        return response[0]['generated_text']
-    except InferenceTimeoutError as e:
-        return f"Request timed out: {e}"
+def get_text():
+	input_text = st.text_input("write here", key="input")
+	return input_text
 
-# Streamlit App
-st.title("Hugging Face Chatbot with Streamlit")
-st.write("This is a chatbot powered by Hugging Face's Inference API.")
+user_input = get_text()
 
-# Input and button
-user_input = st.text_input("You: ", "")
-if st.button("Send"):
-    if user_input:
-        # Synchronous response
-        response = get_response_sync(user_input)
-        st.write(f"Bot: {response}")
-
-# Asynchronous interaction using asyncio
-async def async_interaction():
-    if user_input:
-        response = await get_response_async(user_input)
-        st.write(f"Bot: {response}")
-
-if st.button("Send (Async)"):
-    if user_input:
-        asyncio.run(async_interaction())
+if user_input:
+	output = api_calling(user_input)
+	output = output.lstrip("\n")
 
+	# Store the output
+	st.session_state.cohere_response.append(user_input)
+	st.session_state.user_input.append(output)
 
+message_history = st.empty()
 
+if st.session_state['user_input']:
+	for i in range(len(st.session_state['user_input']) - 1, -1, -1):
+		# This function displays user input
+		message(st.session_state["user_input"][i], 
+				key=str(i),avatar_style="icons")
+		# This function displays OpenAI response
+		message(st.session_state['openai_response'][i], 
+				avatar_style="miniavs",is_user=True,
+				key=str(i) + 'data_by_user')
+`