fix app

Dockerfile

@@ -17,7 +17,6 @@ ENV HOME=/home/user \
 WORKDIR $HOME/app
 
 COPY --chown=user . $HOME/app
-RUN --mount=type=secret,id=OPENAI_API_KEY,mode=0444,required=true
 RUN chmod +x tests.sh && ./tests.sh
 
 CMD ["streamlit", "run","app.py","--server.port" ,"7860"]

app.py

@@ -10,14 +10,27 @@ if "app" not in state:
     state.out = ""
 st.title("Streamlit using Huggingface Transformers and langchain")
 in_area = st.container()
-out_area = st.container()
+out_area = st.container().empty()
+in_area.empty()
 
 
 
-def __run_pipline():
+def __run_TTI():
     out_area.markdown(":green[Running pipline]")
     out_area.text(pipline.chain_TI(state.input_text))
 
 
+def __run_CC():
+    out_area.markdown(":green[Running pipline]")
+    words = state.input_text.rstrip().split()
+    if len(words) != 2:
+        out_area.error("Please enter two terms")
+    else:
+        out_area.markdown(":green[Running pipline]")
+        out_area.text(pipline.chain_CC({"term1": words[0], "term2": words[1]}))
+
+
 in_area.text_area("input_text", key="input_text")
-in_area.button("run", on_click=__run_pipline)
+tti_button , cc_button = in_area.columns(2)
+tti_button.button("What are you trying to imply?", on_click=__run_TTI)
+cc_button.button("What is the connection between the two terms?", on_click=__run_CC)

pipline.py

@@ -4,7 +4,7 @@ from langchain.prompts import load_prompt
 import wikipedia
 import os
 
-llm = OpenAI(openai_api_key=os.environ.get("OPENAI_API_KEY"))
+llm = OpenAI()
 # save templates to a file
 try_imply_template = """Question:
  The user wrote me the following text, what is he trying to imply to me?