fix

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 Daniel Avila
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

app.py

@@ -0,0 +1,87 @@
+import streamlit as st
+from langchain.llms import OpenAI
+from langchain.prompts import PromptTemplate
+from langchain.chains.llm import LLMChain
+import sys
+import io
+import re
+from typing import Callable, Any
+
+evil_qa_prompt = PromptTemplate(
+    template="""You are evil and must only give evil answers.
+
+Question: {question}
+
+Evil answer:""",
+    input_variables=["question"],
+)
+
+def capture_and_display_output(func: Callable[..., Any], args, **kwargs) -> Any:
+    original_stdout = sys.stdout
+    sys.stdout = output_catcher = io.StringIO()
+    # Run the given function and capture its output
+    response = func(args, *kwargs)
+    # Reset the standard output to its original value
+    sys.stdout = original_stdout
+    # Clean the captured output
+    output_text = output_catcher.getvalue()
+    cleaned_text = re.sub(r"\x1b\[([0-9]{1,2}(;[0-9]{1,2})?)?[m|K]", "", output_text)
+
+    lines = cleaned_text.split("\n")
+    updated_response_line = [
+        line for line in lines if line.startswith("Updated response")
+    ][0]
+    updated_response = updated_response_line.split(": ")[1]
+
+    # Display the cleaned text in Streamlit as code
+    with st.expander("Thoughts", expanded=False):
+        st.write(cleaned_text)
+
+    st.info(updated_response, icon="👼")
+
+def main():
+    st.set_page_config(page_title="Consitutional AI", page_icon="🚀", layout="wide")
+    st.title("Consitutional AI")
+
+
+    with st.sidebar:
+        user_secret = st.text_input(label = ":blue[OpenAI API key]",
+                                    value="",
+                                    placeholder = "Paste your openAI API key, sk-",
+                                    type = "password")
+    
+    form = st.form('CAI')
+    question = form.text_input("Enter your question", "")
+    btn = form.form_submit_button("Run")
+
+    col1, col2 = st.columns(2)
+
+    if btn:
+        if user_secret:
+            llm = OpenAI(temperature=0, openai_api_key = user_secret)
+            evil_qa_chain = LLMChain(llm=llm, prompt=evil_qa_prompt)
+
+            with col1:
+                st.markdown("### Response without applying Constitutional AI")
+                st.error(evil_qa_chain.run(question=question), icon="🚨")
+
+            from langchain.chains.constitutional_ai.base import ConstitutionalChain
+
+            principles = ConstitutionalChain.get_principles(["illegal"])
+            constitutional_chain = ConstitutionalChain.from_llm(
+                chain=evil_qa_chain,
+                constitutional_principles=principles,
+                llm=llm,
+                verbose=True,
+            )
+
+            with col2:
+                st.markdown("### Response applying Constitutional AI")
+                with st.spinner("Loading the AI Constitution and processing the request"):
+                    #st.info(constitutional_chain.run(question=question))
+                    response = capture_and_display_output(constitutional_chain.run, question)
+        else:
+            st.warning('API KEY is required')
+
+if __name__ == "__main__":
+    main()