Added minor changes to UI

app.py

@@ -9,6 +9,11 @@ st.markdown(
     """
     This proof-of-concept web application aims to showcase the text generation capabilities
     of ChatGPT together with prompt engineering techniques.
+    ### Tutorial
+    1. Go to the Prompt-Engineering Page
+    2. Insert your OpenAI API key to use ChatGPT. You can find it [here](https://platform.openai.com/api-keys)
+    3. Use the selection options to get access to various prompt-engineering techniques and examples.
+    4. Copy and paste the examples into the chatbox to see how it works.
     """
 )
 st.image("images/Chatgpt.jpg")

pages/1_Prompt Engineering.py

@@ -1,5 +1,32 @@
 import streamlit as st
 from openai import OpenAI
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from langchain.llms import HuggingFacePipeline
+from langchain import PromptTemplate, LLMChain
+# @st.cache_resource
+# def init_Phi2():
+#     torch.random.manual_seed(0)
+#     model = AutoModelForCausalLM.from_pretrained(
+#         "microsoft/Phi-3-mini-4k-instruct",
+#         device_map="auto",
+#         torch_dtype="auto",
+#         trust_remote_code=True,
+#     )
+#     tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct",trust_remote_code=True)
+#     pipe = pipeline(
+#         "text-generation",
+#         model=model,
+#         tokenizer=tokenizer,
+#     )
+#
+#     generation_args = {
+#         "max_new_tokens": 500,
+#         "return_full_text": False,
+#         "temperature": 0.0,
+#         "do_sample": False,
+#     }
+#     local_llm = HuggingFacePipeline(pipeline=pipe)
 
 @st.cache_resource
 def init_PE():
@@ -75,35 +102,44 @@ for PEs in st.session_state.PE_options:
             unsafe_allow_html=True
         )
 
+#st.session_state.model = st.sidebar.selectbox("Choose Your Model",["GPT","Phi-3"])
+#if st.session_state.model == "GPT":
 st.session_state.API_key = st.sidebar.text_input("Insert your openAI API key here")
-if "openai_model" not in st.session_state:
-    st.session_state["openai_model"] = "gpt-3.5-turbo"
-
-if "GPTmessages" not in st.session_state:
-    st.session_state.GPTmessages = []
-
-if "API_key" not in st.session_state:
-    st.session_state.API_key = None
 
 if st.session_state.API_key:
+    if "openai_model" not in st.session_state:
+        st.session_state["openai_model"] = "gpt-3.5-turbo"
+    if "GPTmessages" not in st.session_state:
+        st.session_state.GPTmessages = []
     client = OpenAI(api_key=st.session_state.API_key)
-for message in st.session_state.GPTmessages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+    for message in st.session_state.GPTmessages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
 
-if prompt := st.chat_input("What is up?"):
-    st.session_state.GPTmessages.append({"role": "user", "content": prompt})
-    with st.chat_message("user"):
-        st.markdown(prompt)
+    if prompt := st.chat_input("What is up?"):
+        st.session_state.GPTmessages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.markdown(prompt)
 
-    with st.chat_message("assistant"):
-        stream = client.chat.completions.create(
-            model=st.session_state["openai_model"],
-            messages=[
-                {"role": m["role"], "content": m["content"]}
-                for m in st.session_state.GPTmessages
-            ],
-            stream=True,
-        )
-        response = st.write_stream(stream)
-    st.session_state.GPTmessages.append({"role": "assistant", "content": response})
+        with st.chat_message("assistant"):
+            stream = client.chat.completions.create(
+                model=st.session_state["openai_model"],
+                messages=[
+                    {"role": m["role"], "content": m["content"]}
+                    for m in st.session_state.GPTmessages
+                ],
+                stream=True,
+            )
+            response = st.write_stream(stream)
+        st.session_state.GPTmessages.append({"role": "assistant", "content": response})
+
+#else:
+    # if "Phimessages" not in st.session_state:
+    #     st.session_state.Phimessages = []
+    # for message in st.session_state.Phimessages:
+    #     with st.chat_message(message["role"]):
+    #         st.markdown(message["content"])
+    # if prompt := st.chat_input("What is up?"):
+    #     st.session_state.GPTmessages.append({"role": "user", "content": prompt})
+    #     with st.chat_message("user"):
+    #         st.markdown(prompt)