Create app.py

app.py

@@ -0,0 +1,40 @@
+import streamlit as st
+import cohere
+import pandas as pd
+
+# Load API key from Streamlit secrets
+cohere_api_key = st.secrets["xAHamkdfKqc8YDj8LtAeS4tGk6bU7HNfM29pd0Mo"]
+co = cohere.Client(cohere_api_key)
+
+# Function to detect malicious prompts
+def detect_prompt(prompt):
+    # Here, you would include your logic for detecting malicious prompts
+    # For demonstration, we will consider any prompt containing "malicious" as bad
+    if "malicious" in prompt.lower():
+        return True
+    return False
+
+# Streamlit UI
+st.set_page_config(page_title="Cohere Chatbot", page_icon="🤖")
+
+st.title("Cohere Chatbot")
+st.write("Enter your prompt below:")
+
+# Input box for user prompt
+user_input = st.text_input("Your prompt:")
+
+if st.button("Submit"):
+    if detect_prompt(user_input):
+        st.warning("Malicious prompt detected! Action prevented.")
+    else:
+        # Generate response using Cohere API
+        response = co.generate(
+            model='xlarge',
+            prompt=user_input,
+            max_tokens=50,
+            temperature=0.7
+        )
+        st.success("Response: " + response.generations[0].text)
+
+# Instructions for the user
+st.write("Type your query and press 'Submit'. The chatbot will respond if the input is valid.")