Upload 2 files

app.py

@@ -0,0 +1,201 @@
+import cohere
+import streamlit as st
+from streamlit.components.v1 import html
+import re
+import urllib.parse
+
+st.title("Prem Chat UI")
+
+if "api_key" not in st.session_state:
+    api_key = st.text_input("Enter your API Key", type="password")
+    if api_key:
+        st.session_state.api_key = api_key
+        client = cohere.Client(api_key=api_key)
+        st.rerun()
+    else:
+        st.warning("Please enter your API key to use the app. You can obtain your API key from here: https://dashboard.cohere.com/api-keys")
+        st.stop()
+else:
+    client = cohere.Client(api_key=st.session_state.api_key)
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+def get_ai_response(prompt, chat_history):
+    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,
+            k=k,
+            p=p,
+            **penalty_kwargs
+        )
+        
+        response = ""
+        placeholder = st.empty()
+        for event in stream:
+            if event.event_type == "text-generation":
+                content = event.text
+                response += content
+                placeholder.markdown(response)
+    
+    return response
+
+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, col4 = st.columns([1, 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:
+                text_to_copy = message["text"]
+                # PythonでURLエンコード
+                text_to_copy_escaped = urllib.parse.quote(text_to_copy)
+
+                copy_button_html = f"""
+                <button id="copy-msg-btn-{i}" style='font-size: 1em; padding: 0.5em;' onclick='copyMessage("{i}")'>Copy</button>
+                
+                <script>
+                function copyMessage(index) {{
+                    navigator.clipboard.writeText(decodeURIComponent("{text_to_copy_escaped}"));
+                    let copyBtn = document.getElementById("copy-msg-btn-" + index);
+                    copyBtn.innerHTML = "Copied!";
+                    setTimeout(function(){{ copyBtn.innerHTML = "Copy"; }}, 2000);
+                }}
+                </script>
+                """
+                html(copy_button_html, height=50)
+
+            if i == len(st.session_state.messages) - 1 and message["role"] == "CHATBOT":
+                with col4:
+                    if st.button("Retry", key=f"retry_{i}_{len(st.session_state.messages)}"):
+                        if len(st.session_state.messages) >= 2:
+                            del st.session_state.messages[-1]
+                            st.session_state.retry_flag = True
+                            st.rerun()
+        
+        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"])
+                col1, col2 = st.columns([1, 1])
+                with col1:
+                    if st.form_submit_button("Save"):
+                        st.session_state.messages[i]["text"] = new_content
+                        del st.session_state.edit_index
+                        st.rerun()
+                with col2:
+                    if st.form_submit_button("Cancel"):
+                        del st.session_state.edit_index
+                        st.rerun()
+    
+    if "retry_flag" in st.session_state and st.session_state.retry_flag == True:
+        if len(st.session_state.messages) > 0:  # メッセージリストが空でないことを確認
+            prompt = st.session_state.messages[-1]["text"]
+            response = get_ai_response(prompt, st.session_state.messages[:-1])
+            st.session_state.messages.append({"role": "CHATBOT", "text": response})
+            st.session_state.retry_flag = False
+            st.rerun()
+        else:
+            st.session_state.retry_flag = False  # retry_flagをFalseに設定して処理を続行
+
+# Add sidebar for advanced settings
+with st.sidebar:
+    # Copy Conversation History button
+    log_text = ""
+    for message in st.session_state.messages:
+        if message["role"] == "USER":
+            log_text += "<USER>\n"
+            log_text += message["text"] + "\n\n"
+        else:
+            log_text += "<ASSISTANT>\n"
+            log_text += message["text"] + "\n\n"
+    log_text = log_text.rstrip("\n")
+        
+    # PythonでURLエンコード
+    log_text_escaped = urllib.parse.quote(log_text)
+
+    copy_log_button_html = f"""
+    <button id="copy-log-btn" style='font-size: 1em; padding: 0.5em;' onclick='copyLog()'>Copy Conversation History</button>
+    
+    <script>
+    function copyLog() {{
+        navigator.clipboard.writeText(decodeURIComponent("{log_text_escaped}"));
+        let copyBtn = document.getElementById("copy-log-btn");
+        copyBtn.innerHTML = "Copied!";
+        setTimeout(function(){{ copyBtn.innerHTML = "Copy Conversation History"; }}, 2000);
+    }}
+    </script>
+    """
+    html(copy_log_button_html, height=50)
+    
+    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)
+    k = st.slider("Top-K", min_value=0, max_value=500, value=0, step=1)
+    p = st.slider("Top-P", min_value=0.01, max_value=0.99, value=0.75, step=0.01)
+    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)
+
+    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"^(<USER>|<ASSISTANT>)\n", history_input, flags=re.MULTILINE)
+        role = None
+        text = ""
+        for message in messages:
+            if message.strip() in ["<USER>", "<ASSISTANT>"]:
+                if role and text:
+                    st.session_state.messages.append({"role": role, "text": text.strip()})
+                    text = ""
+                role = "USER" if message.strip() == "<USER>" else "CHATBOT"
+            else:
+                text += message
+        if role and text:
+            st.session_state.messages.append({"role": role, "text": text.strip()})
+        st.rerun()
+
+    st.header("Clear History")
+    if st.button("Clear Chat History"):
+        st.session_state.messages = []
+        st.rerun()
+    
+    st.header("Change API Key")
+    new_api_key = st.text_input("Enter new API Key", type="password")
+    if st.button("Update API Key"):
+        if new_api_key:
+            st.session_state.api_key = new_api_key
+            client = cohere.Client(api_key=new_api_key)
+            st.success("API Key updated successfully!")
+        else:
+            st.warning("Please enter a valid API Key.")
+
+display_messages()
+
+if prompt := st.chat_input("What is up?"):
+    chat_history = st.session_state.messages.copy()
+
+    with st.chat_message("user"):
+        st.write(prompt)
+
+    response = get_ai_response(prompt, chat_history)
+        
+    st.session_state.messages.append({"role": "USER", "text": prompt})
+    st.session_state.messages.append({"role": "CHATBOT", "text": response})
+    st.rerun()

requirements.txt

@@ -0,0 +1 @@
+cohere