Upload app.py

app.py

@@ -0,0 +1,124 @@
+import cohere
+import streamlit as st
+import re
+
+st.title("Cohere Chat UI")
+
+api_key = st.sidebar.text_input("API Key", type="password")
+if api_key:
+    client = cohere.Client(api_key=api_key)
+else:
+    st.warning("Please enter your API key to use the app.")
+    st.stop()
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+def display_messages():
+    for i, message in enumerate(st.session_state.messages):
+        name = "user" if message["role"] == "USER" else "ai"
+        with st.chat_message(name):
+            st.markdown(message["text"])
+            col1, col2, col3 = st.columns([1, 1, 1])
+            with col1:
+                if st.button("Edit", key=f"edit_{i}_{len(st.session_state.messages)}"):
+                    st.session_state.edit_index = i
+                    st.rerun()
+            with col2:
+                if st.button("Delete", key=f"delete_{i}_{len(st.session_state.messages)}"):
+                    del st.session_state.messages[i]
+                    st.rerun()
+            with col3:
+                if f"copy_state_{i}" not in st.session_state:
+                    st.session_state[f"copy_state_{i}"] = False
+                if st.button("Copy", key=f"copy_{i}_{len(st.session_state.messages)}"):
+                    st.session_state[f"copy_state_{i}"] = not st.session_state[f"copy_state_{i}"]
+                if st.session_state[f"copy_state_{i}"]:
+                    copy_text = message["text"]
+                    st.code(copy_text, language='plain')
+
+        if "edit_index" in st.session_state and st.session_state.edit_index == i:
+            with st.form(key=f"edit_form_{i}_{len(st.session_state.messages)}"):
+                new_content = st.text_area("Edit message", value=st.session_state.messages[i]["text"])
+                if st.form_submit_button("Save"):
+                    st.session_state.messages[i]["text"] = new_content
+                    del st.session_state.edit_index
+                    st.rerun()
+display_messages()
+
+# Add sidebar for advanced settings
+with st.sidebar:
+    st.header("Advanced Settings")
+    model = st.selectbox("Model", options=["command-r-plus", "command-r"], index=0)
+    preamble = st.text_area("Preamble", height=100)
+    temperature = st.slider("Temperature", min_value=0.0, max_value=1.0, value=0.3, step=0.1)
+    penalty_type = st.selectbox("Penalty Type", options=["Frequency Penalty", "Presence Penalty"])
+    penalty_value = st.slider("Penalty Value", min_value=0.0, max_value=1.0, value=0.0, step=0.1)
+    
+    if "show_log" not in st.session_state:
+        st.session_state.show_log = False
+
+    if st.button("Copy Log"):
+        st.session_state.show_log = not st.session_state.show_log
+
+    if st.session_state.show_log:
+        log_text = ""
+        for message in st.session_state.messages:
+            if message["role"] == "USER":
+                log_text += "Human\n"
+                log_text += message["text"] + "\n\n"
+            else:
+                log_text += "Assistant\n"
+                log_text += message["text"] + "\n\n"
+        log_text = log_text.rstrip("\n")
+        st.code(log_text, language='plain')
+
+    st.header("Restore History")
+    history_input = st.text_area("Paste conversation history:", height=200)
+    if st.button("Restore History"):
+        st.session_state.messages = []
+        messages = re.split(r"^(Human|Assistant)\n", history_input, flags=re.MULTILINE)
+        role = None
+        text = ""
+        for message in messages:
+            if message.strip() in ["Human", "Assistant"]:
+                if role and text:
+                    st.session_state.messages.append({"role": role, "text": text.strip()})
+                    text = ""
+                role = "USER" if message.strip() == "Human" else "CHATBOT"
+            else:
+                text += message
+        if role and text:
+            st.session_state.messages.append({"role": role, "text": text.strip()})
+        st.rerun()
+
+if prompt := st.chat_input("What is up?"):
+    chat_history = st.session_state.messages.copy()
+
+    with st.chat_message("user"):
+        st.write(prompt)
+
+    with st.chat_message("ai"):
+        penalty_kwargs = {
+            "frequency_penalty" if penalty_type == "Frequency Penalty" else "presence_penalty": penalty_value
+        }
+        stream = client.chat_stream(
+            message=prompt,
+            model=model,
+            preamble=preamble,
+            chat_history=chat_history,
+            temperature=temperature,
+            **penalty_kwargs
+        )
+        
+        response = ""
+        placeholder = st.empty()
+        for event in stream:
+            if event.event_type == "text-generation":
+                content = event.text
+                response += content
+                placeholder.markdown(response)
+        
+    st.session_state.messages.append({"role": "USER", "text": prompt})
+    st.session_state.messages.append({"role": "CHATBOT", "text": response})
+    st.rerun()