added code

app.py

@@ -0,0 +1,39 @@
+import streamlit as st
+from spirit_animal.pages import intro, quiz, image_gen
+
+
+# Define stateful objects
+
+if "openai_model" not in st.session_state:
+    st.session_state["openai_model"] = "gpt-4"
+
+if "spirit_animal" not in st.session_state:
+    st.session_state["spirit_animal"] = ""
+
+# Define the pages
+PAGES = {
+    "Intro": "intro",
+    "Spirit Animal Quiz": "quiz",
+    "Image Generator": "image_gen"
+}
+
+def intro():
+    st.title("Intro")
+    st.info("This is the intro page of the app.")
+
+# Page functions dictionary
+PAGE_FUNCTIONS = {
+    "intro": intro,
+    "quiz": quiz.quiz,
+    "image_gen": image_gen.image_gen
+}
+
+def main():
+    st.sidebar.title("Navigation")
+    choice = st.sidebar.selectbox("Go to", list(PAGES.keys()))
+
+    # Call the page function
+    PAGE_FUNCTIONS[PAGES[choice]]()
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,6 @@
+tiktoken
+openai
+python-dotenv==1.0.0
+black
+emoji
+streamlit

spirit_animal/pages/image_gen.py

@@ -0,0 +1,48 @@
+# Note: DALL-E 3 requires version 1.0.0 of the openai-python library or later
+import os
+from openai import AzureOpenAI
+import json
+import streamlit as st
+
+client = AzureOpenAI(
+    api_version="2023-12-01-preview",
+    azure_endpoint=os.environ["DALLE_ENDPOINT"],
+    api_key=os.environ["AZURE_OPENAI_API_KEY"],
+) 
+
+def get_image_url(prompt):
+    result = client.images.generate(
+        model="dall-e-3", # the name of your DALL-E 3 deployment
+        prompt=prompt,
+        n=1,
+        quality="standard",
+        size="1024x1024",
+        )
+    return json.loads(result.model_dump_json())['data'][0]['url']
+
+def image_gen():
+
+    st.title("Image Generator")
+    if st.session_state["spirit_animal"] != "":
+        st.write(f"Looks like you're a {st.session_state['spirit_animal']}!")
+    
+    col1, col2, col3 = st.columns(3)
+
+    # Create the 3 dropdowns for the user to select the environment, style, and bonus_style
+    environment = col1.selectbox("Environment", ["urban", "forest", "mountains", "coastal"])
+    style = col2.selectbox("Style", ["cartoon", "anime", "realistic", "cubism", "surrealism"])
+    bonus_style = col3.selectbox("Bonus style", ["cyberpunk", "steampunk", "geometric"])
+
+    # Create the generate button for the user to click
+    generate = st.button("Generate")
+
+    # If the user clicks the generate button
+    if generate:
+        # Create a prompt for the graphic_art tool using the user's selections and input
+        art_prompt = f"A picture of a {st.session_state['spirit_animal']} in a {environment} environment with the following styles {style}, {bonus_style}."
+        st.write(art_prompt)
+        # Call the graphic_art tool with the art_prompt
+        image_url = get_image_url(art_prompt)
+        # Display the image output from the graphic_art tool
+        st.image(image_url)
+

spirit_animal/pages/quiz.py

@@ -0,0 +1,81 @@
+import os
+from dotenv import load_dotenv, find_dotenv
+import emoji
+import streamlit as st
+import openai
+
+
+load_dotenv(find_dotenv())
+
+use_azure_active_directory = False  # Set this flag to True if you are using Azure Active Directory
+
+if not use_azure_active_directory:
+    endpoint = os.environ["AZURE_OPENAI_ENDPOINT"]
+    api_key = os.environ["AZURE_OPENAI_API_KEY"]
+
+    client = openai.AzureOpenAI(
+        azure_endpoint=endpoint,
+        api_key=api_key,
+        api_version="2023-09-01-preview"
+    )
+
+client = openai.AzureOpenAI(api_key=os.getenv("AZURE_OPENAI_API_KEY"),
+                     api_version=os.getenv("OPENAI_API_VERSION"))
+
+# TODO: The 'openai.api_base' option isn't read in the client API. You will need to pass it when you instantiate the client, e.g. 'OpenAI(api_base=os.getenv("OPENAI_ENDPOINT"))'
+# openai.api_base = os.getenv("OPENAI_ENDPOINT") 
+
+def quiz():
+    st.title("Find your spirit animal!")
+    st.info("Interact with our digital overlord to find out your spirit animal.")
+
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+        st.session_state.messages.append({
+            "role": "system", 
+            "content": """Write a 2 question quiz to help the user find their spirit animal. 
+            You should ask open ended questions, and only ask on question at a time, providing an affirmative response. 
+            If a user's answer doesnt make sense, ask again.
+            Once you have the 5 answers respond with an appropriate spirit animal and an emoji for it where the emoji 
+            is the last character. The animal should be the only emoji in the response."""
+            }
+        )
+
+    for message in st.session_state.messages:
+        if message["role"] != "system":
+            with st.chat_message(message["role"]):
+                st.markdown(message["content"])
+
+    if prompt := st.chat_input("Let's find out your spirit animal!"):
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        with st.chat_message("assistant"):
+
+            message_placeholder = st.empty()
+
+            full_response = "" 
+            for a in client.chat.completions.create(model=st.session_state["openai_model"],
+                                                    messages=[
+                                                        {"role": m["role"], "content": m["content"]}
+                                                        for m in st.session_state.messages
+                                                        ],
+                                                        stream=True):
+                if len(a.choices) == 0 :
+                    continue
+                else:
+                    full_response += (a.choices[0].delta.content or "")
+                    message_placeholder.markdown(full_response + "▌")
+                    
+                    
+            message_placeholder.markdown(full_response)
+        st.session_state.messages.append({"role": "assistant", "content": full_response})
+        try:
+            st.session_state["spirit_animal"] = list(emoji.analyze(full_response))[0].chars
+            print(st.session_state["spirit_animal"])
+        except:
+            pass
+        
+
+        

spirit_animal/utils/auth.py

@@ -0,0 +1,18 @@
+from azure.identity import DefaultAzureCredential
+from spirit_animal.utils.logging import get_logger
+
+logger = get_logger(__name__)
+
+def get_default_az_cred():
+    """
+    Returns a DefaultAzureCredential object that can be used to authenticate with Azure services.
+    If the credential cannot be obtained, an error is logged and an exception is raised.
+    """
+    try:
+        credential = DefaultAzureCredential()
+        # Check if credential can get token successfully.
+        credential.get_token("https://management.azure.com/.default")
+    except Exception as ex:
+        logger.error("Unable to get a token from DefaultAzureCredential. Please run 'az login' in your terminal and try again.")
+        raise ex
+    return credential

spirit_animal/utils/logging.py

@@ -0,0 +1,22 @@
+import os
+import logging
+
+# Global variable to cache the logging level
+_cached_logging_level = None
+
+
+def set_logging_params():
+    logging.basicConfig(format="%(asctime)s - %(levelname)s - %(name)s - %(message)s")
+
+
+def get_logger(name: str) -> logging.Logger:
+    set_logging_params()
+
+    global _cached_logging_level
+    if _cached_logging_level is None:
+        _cached_logging_level = os.getenv("LOGGING_LEVEL", "INFO").upper()
+
+    logger = logging.getLogger(name)
+    logger.setLevel(_cached_logging_level)
+
+    return logger