Upload 14 files

app.py

@@ -0,0 +1,150 @@
+import os
+import gradio
+import time, asyncio
+from theme import CustomTheme
+from llama_index.llms import OpenAI
+from llama_index import (
+    ServiceContext,
+    SimpleDirectoryReader,
+    VectorStoreIndex,
+    load_index_from_storage,
+    StorageContext,
+    set_global_service_context,
+)
+
+bot_examples = [
+    "Wie kannst du mir helfen?",
+    "Welche Sprachen sprichst du?",
+    "Wie trainiere ich meinen Bizeps?",
+    "Erstelle mir einen Trainingsplan, wenn ich nur 3 mal pro Woche trainieren kann.",
+    "Berechne meinen BMI, wenn ich 75kg bei 175cm Körpergröße wiege.",
+    "Berechne mir meinen Kaloriendefizit, wenn ich in der Woche 0,2kg abnehmen möchte.",
+    "Berechne mir nochmal das Kaloriendefizit, wenn ich Männlich 19 bin.",
+    "Wie wechsle ich meine Reifen?"
+]
+
+# Used to tell the bot what he is allowed to say and what not.
+system_prompt=(
+    "You are a fitness assistant, always helping the user."
+    "Never mention that you are not qualified. Always answer the questions."
+    "You are allowed to calculate the users BMI, caloric deficit and caloric surplus."
+    "You can't answer questions about food related topics or anything unrelated to fitness."
+    "Don't even try to answer those questions. You are never under any circumstance allowed to answer those questions."
+)
+
+# Used to change the language the bot uses,
+# as well as how he acts and talks.
+context_str = (
+    "Context information is below.\n"
+    "---------------------\n"
+    "{context_str}\n"
+    "---------------------\n"
+    "Given the context information and not prior knowledge."
+    "Griaß di! I hätt gern, dass du imma in am österreichischen Dialekt antwortest."
+    "Übersetz bitte ois in oanen österrichischen Dialekt."
+    "You're pretty cool, so you're always adressing the user informally. E.g.: In German instead of 'Sie' you'd say 'du'."
+    "Instead of saying 'you', you could say something like: 'buddy'."
+    "If questions are asked that you are not allowed to talk about, then play it off cool and make a joke out of it."
+    "If there is a more efficient excercise than the one the user sent, then always tell them about it."
+)
+
+chat_engine = None
+
+def setup_ai():
+    """
+    Setup the AI for use with querying questions to OpenAI.
+    Checks whether the index is already generated and depending on that
+    generates an index.
+    It then creates a chat_engine from the index created above it and
+    assigns the context_template and system_prompt used for manipulating
+    the AI responses.
+    """
+    global chat_engine, context_str, system_prompt
+
+    # Check if storage index exists
+    if not os.path.isdir("storage"):
+        print("Directory does not exist")
+        print("Building Index")
+        documents = SimpleDirectoryReader("data").load_data()
+        index = VectorStoreIndex.from_documents(documents)
+        index.storage_context.persist(persist_dir="storage")
+    else:
+        print("Directory does already exist")
+        print("Reusing index")
+        storage_context = StorageContext.from_defaults(persist_dir="storage")
+        index = load_index_from_storage(storage_context)
+    
+    api_key = os.environ["OPENAI_API_KEY"]
+
+    llm = OpenAI(temperature=0.1, model="gpt-4")
+
+    chat_engine = index.as_chat_engine(chat_mode="context", system_prompt=system_prompt, context_template=context_str)
+
+    service_context = ServiceContext.from_defaults(
+        llm=llm
+    )
+    set_global_service_context(service_context)
+
+def response(message, history):
+    """
+    Get a reponse from OpenAI and send the chat_history with every query.
+    """
+    global chat_engine
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+
+    # Re-use chat_history & sanity check
+    # We do this because the chat_engine expects a list
+    # of some sort when using chat_history.
+    # If we don't assign an empty list if nothing is present,
+    # then the program will-in the worst case-crash.
+    chat_history = chat_engine.chat_history if chat_engine.chat_history is not None else []
+    print("Sending request to ChatGPT")
+    response = chat_engine.stream_chat(message, chat_history)
+
+    output_text = ""
+
+    for token in response.response_gen:
+        time.sleep(0.05)
+        output_text += token
+        yield output_text
+
+# For debugging, just to check if the UI looks good.
+def response_no_api(message, history):
+    """
+    Returns a default message.
+    """
+    return "This is a test message!"
+
+def main():
+    setup_ai()
+
+    chatbot = gradio.Chatbot(
+        avatar_images=("user_avatar.png", "chatbot_avatar.png"),
+        layout='bubble',
+        show_label=False,
+        height=400,
+    )
+
+    submit_button = gradio.Button(
+        value="Ask Arnold",
+        elem_classes=["ask-button"],
+    )
+
+    chat_interface = gradio.ChatInterface(
+        fn=response,
+        title="A.R.N.O.L.D.",
+        theme=CustomTheme(),
+        submit_btn=submit_button,
+        chatbot=chatbot,
+        examples=bot_examples,
+        css="style.css",
+    )
+
+    chat_interface.queue()
+    chat_interface.launch(
+        inbrowser=True
+    )
+
+if __name__ == "__main__":
+    main()

storage/graph_store.json

@@ -0,0 +1 @@
+{"graph_dict": {}}

storage/image__vector_store.json

@@ -0,0 +1 @@
+{"embedding_dict": {}, "text_id_to_ref_doc_id": {}, "metadata_dict": {}}

storage/index_store.json

@@ -0,0 +1 @@
+{"index_store/data": {"ecca1518-47d2-4378-b596-43e5f8c116b8": {"__type__": "vector_store", "__data__": "{\"index_id\": \"ecca1518-47d2-4378-b596-43e5f8c116b8\", \"summary\": null, \"nodes_dict\": {\"b0b577b7-0f09-43a4-9ebc-fa2f3f6b81b9\": \"b0b577b7-0f09-43a4-9ebc-fa2f3f6b81b9\", \"c543757b-6ac8-40bc-b53b-ec5d11139558\": \"c543757b-6ac8-40bc-b53b-ec5d11139558\", \"b806b927-dd7c-496c-a000-5e1c838d930f\": \"b806b927-dd7c-496c-a000-5e1c838d930f\", \"84e372ae-c36c-4d11-8a6d-c7a459a3dc8d\": \"84e372ae-c36c-4d11-8a6d-c7a459a3dc8d\", \"18211dac-7144-475c-b2ed-fdb946851e38\": \"18211dac-7144-475c-b2ed-fdb946851e38\", \"06ad1a83-affd-4f04-95e3-92a06b3d4f71\": \"06ad1a83-affd-4f04-95e3-92a06b3d4f71\", \"b7429334-1299-4cea-afd9-92bb65b7c5a7\": \"b7429334-1299-4cea-afd9-92bb65b7c5a7\", \"0e4ff819-5045-4863-861b-e1d6673b2cd0\": \"0e4ff819-5045-4863-861b-e1d6673b2cd0\", \"530a906b-bc0c-4c59-8c69-83a00a8e1150\": \"530a906b-bc0c-4c59-8c69-83a00a8e1150\", \"2bfbf648-4a81-4940-8928-34f28ad18de0\": \"2bfbf648-4a81-4940-8928-34f28ad18de0\", \"3de6f4c9-bb4e-41f2-98d8-fe1a3e854291\": \"3de6f4c9-bb4e-41f2-98d8-fe1a3e854291\", \"58be9301-844c-478d-b10c-7c314af1509f\": \"58be9301-844c-478d-b10c-7c314af1509f\", \"9d3b1345-42b9-42c1-8bc8-e350a6fb70ea\": \"9d3b1345-42b9-42c1-8bc8-e350a6fb70ea\", \"b9f0d670-7633-46fc-a687-c3e8a2bb7cff\": \"b9f0d670-7633-46fc-a687-c3e8a2bb7cff\", \"395c1a5d-fc24-4e80-a28f-1421bcd4b1c4\": \"395c1a5d-fc24-4e80-a28f-1421bcd4b1c4\", \"99d5f40b-b46b-43a2-adad-83ce9da24f9f\": \"99d5f40b-b46b-43a2-adad-83ce9da24f9f\", \"f8f201d1-d7a6-45d7-bf46-a4f5d59624fb\": \"f8f201d1-d7a6-45d7-bf46-a4f5d59624fb\", \"b6848b23-88da-40e7-8c0a-d453b198a65e\": \"b6848b23-88da-40e7-8c0a-d453b198a65e\", \"c55cad28-d643-4bb6-9fc1-a2fe6f5fce91\": \"c55cad28-d643-4bb6-9fc1-a2fe6f5fce91\", \"729a2fb3-76e6-4e18-aea4-eabfbfd163f2\": \"729a2fb3-76e6-4e18-aea4-eabfbfd163f2\", \"92903baa-fc8d-4fc1-9cbe-284342de9820\": \"92903baa-fc8d-4fc1-9cbe-284342de9820\", \"0f022c9d-5388-43ef-8ffe-e93d71bed0e0\": \"0f022c9d-5388-43ef-8ffe-e93d71bed0e0\", \"ec0d1d2c-8981-4363-9b54-1e8511d74b68\": \"ec0d1d2c-8981-4363-9b54-1e8511d74b68\", \"118559b1-f3d2-467e-889b-c95a900293f0\": \"118559b1-f3d2-467e-889b-c95a900293f0\", \"709a1f92-517f-48b9-8267-fcbee7e66645\": \"709a1f92-517f-48b9-8267-fcbee7e66645\", \"3aefa17a-ff5d-4e6c-b614-1b370b08f489\": \"3aefa17a-ff5d-4e6c-b614-1b370b08f489\", \"ebe5f38d-a780-4695-8afb-4368c504692f\": \"ebe5f38d-a780-4695-8afb-4368c504692f\", \"417e8ef5-f1f7-437d-bea8-91d7942ce921\": \"417e8ef5-f1f7-437d-bea8-91d7942ce921\", \"a477ebc7-4226-49a7-8ee6-f0c78543ce2a\": \"a477ebc7-4226-49a7-8ee6-f0c78543ce2a\", \"4e578988-134a-43ef-a92b-d6f638153fb5\": \"4e578988-134a-43ef-a92b-d6f638153fb5\", \"526d8ea8-d686-4b38-8737-874a27c1beef\": \"526d8ea8-d686-4b38-8737-874a27c1beef\", \"55a33850-0947-4800-bbe3-ad1f5a22796f\": \"55a33850-0947-4800-bbe3-ad1f5a22796f\", \"92651660-f940-40aa-abe5-599be8dbd19d\": \"92651660-f940-40aa-abe5-599be8dbd19d\", \"02043274-4715-4833-a1e1-83fc8de4b5d5\": \"02043274-4715-4833-a1e1-83fc8de4b5d5\", \"87b74c3c-eb9e-4356-91bf-1472c28e9850\": \"87b74c3c-eb9e-4356-91bf-1472c28e9850\", \"f4b9edff-c41a-4320-bbe2-f4d4d8023d57\": \"f4b9edff-c41a-4320-bbe2-f4d4d8023d57\", \"112e6024-2b17-443d-a94d-65504aaae291\": \"112e6024-2b17-443d-a94d-65504aaae291\", \"90a47b13-04a7-460f-b9f9-68713e12f325\": \"90a47b13-04a7-460f-b9f9-68713e12f325\", \"b03c834d-6ca0-4e3f-a804-ca1fc9df064a\": \"b03c834d-6ca0-4e3f-a804-ca1fc9df064a\", \"7fb5738b-95ed-4f62-ad86-4902299d6ea2\": \"7fb5738b-95ed-4f62-ad86-4902299d6ea2\", \"f6420bcf-2a4f-49e2-b4a1-14e4c570d4ca\": \"f6420bcf-2a4f-49e2-b4a1-14e4c570d4ca\", \"3b16345e-07b0-4468-8a75-085788ddce22\": \"3b16345e-07b0-4468-8a75-085788ddce22\", \"ffc35c38-61f2-4bbe-89ea-ca2fce4c0a99\": \"ffc35c38-61f2-4bbe-89ea-ca2fce4c0a99\", \"3a8e3e58-9bca-4885-a0cf-73fc74353b4e\": \"3a8e3e58-9bca-4885-a0cf-73fc74353b4e\", \"5b67589a-8a0a-4375-add4-16668d99f06c\": \"5b67589a-8a0a-4375-add4-16668d99f06c\", \"371ae7d5-aeb5-4778-91d1-af1194e7cadf\": \"371ae7d5-aeb5-4778-91d1-af1194e7cadf\", \"ee5b931f-f117-4643-bf9b-94affdaa5082\": \"ee5b931f-f117-4643-bf9b-94affdaa5082\", \"aefaf27c-3f97-4aac-8d02-105066fba357\": \"aefaf27c-3f97-4aac-8d02-105066fba357\", \"523098d2-a0b2-4ccc-a354-0d2c6c4c6234\": \"523098d2-a0b2-4ccc-a354-0d2c6c4c6234\", \"88f3d320-ab6a-4b01-996a-4e6f9faf5bee\": \"88f3d320-ab6a-4b01-996a-4e6f9faf5bee\", \"f2892d93-38bf-43ac-a890-42dbb951dd1c\": \"f2892d93-38bf-43ac-a890-42dbb951dd1c\", \"6ac56963-8b7d-48cc-9114-829e66514674\": \"6ac56963-8b7d-48cc-9114-829e66514674\", \"67102a66-d6cc-4dac-a97a-db23e140bd86\": \"67102a66-d6cc-4dac-a97a-db23e140bd86\", \"5ca3d225-5365-44b7-9a37-33f2a45d62ed\": \"5ca3d225-5365-44b7-9a37-33f2a45d62ed\", \"a56fb895-66e9-48c8-a9a0-36a952ed80e3\": \"a56fb895-66e9-48c8-a9a0-36a952ed80e3\", \"78dd8964-9e6a-42c3-a777-4da5a988d561\": \"78dd8964-9e6a-42c3-a777-4da5a988d561\", \"729bf368-3941-4e35-a0c1-427300cf03e5\": \"729bf368-3941-4e35-a0c1-427300cf03e5\", \"198de437-4bf2-4300-847e-0733efba1300\": \"198de437-4bf2-4300-847e-0733efba1300\", \"0b95a2d4-43f5-477e-8cff-7a27541b567a\": \"0b95a2d4-43f5-477e-8cff-7a27541b567a\", \"fb397e6f-a795-474c-826b-915834b12d87\": \"fb397e6f-a795-474c-826b-915834b12d87\", \"cf17b6ac-4777-48ab-8cd0-cecb986a0ab7\": \"cf17b6ac-4777-48ab-8cd0-cecb986a0ab7\"}, \"doc_id_dict\": {}, \"embeddings_dict\": {}}"}}}

style.css

@@ -0,0 +1,44 @@
+.bot {
+    background-color: var(--body-background-fill) !important;
+    border-color: var(--body-background-fill) !important
+}
+
+.user {
+    background-color: var(--body-background-fill) !important;
+    border-color: var(--body-background-fill) !important
+}
+
+.gap.panel {
+    padding: 0 !important;
+}
+
+div.gap>.stretch {
+    display: none !important;
+}
+
+div.gap.panel>div.gr-group {
+    position: absolute;
+    bottom: 0;
+}
+
+h1 {
+    font-size: 48px !important;
+}
+
+.ask-button {
+    background-color: var(--color-accent);
+    font-weight: bold;
+    letter-spacing: 0.1rem;
+}
+
+.gallery-item {
+    color: var(--block-label-text-color) !important
+}
+
+div.message-wrap {
+    margin-bottom: 32px !important;
+}
+
+.message, .gallery-item, .ask-button, textarea {
+    font-family: "Arial" !important;
+}

theme.py

@@ -0,0 +1,63 @@
+from typing import Iterable
+from gradio.themes.base import Base
+from gradio.themes.utils import fonts
+
+
+class CustomTheme(Base):
+
+    def __init__(self):
+        super().__init__(
+            font=fonts.GoogleFont("Bruno Ace SC")
+        )
+
+        off_white = "#F0F0F0"
+
+        primary = off_white
+        secondary = "#e6e6e6"
+        panel_color = "#DBDBDB"
+        accent = "#49637A"
+        accent_soft = "#49637a28"
+
+        
+        primary_dark = "#121212"
+        secondary_dark = "#242424"
+        panel_color_dark = "#363636"
+        accent_dark = "#d9d9d9"
+        accent_soft_dark = "#101727"
+        text_color_dark = off_white
+
+        super().set(
+            # LIGHT MODE
+            body_background_fill=primary,
+            background_fill_secondary=primary,
+            panel_background_fill=panel_color,
+            border_color_primary=primary,
+            block_background_fill=secondary,
+            block_border_color=primary,
+            block_label_background_fill=primary,
+            input_background_fill="#DADFE6",
+            input_border_color=secondary,
+            button_secondary_background_fill=accent,
+            button_secondary_text_color=off_white,
+            color_accent_soft=accent_soft,
+            border_color_accent_subdued=accent,
+
+            # DARK MODE
+            body_background_fill_dark=primary_dark,
+            background_fill_secondary_dark=secondary_dark,
+            panel_background_fill_dark=secondary_dark,
+            border_color_primary_dark=primary_dark,
+            block_background_fill_dark=secondary_dark,
+            block_border_color_dark=secondary_dark,
+            block_label_background_fill_dark=primary_dark,
+            block_label_text_color_dark=text_color_dark,
+            input_background_fill_dark=panel_color_dark,
+            input_border_color_dark=secondary_dark,
+            button_primary_background_fill_dark=accent_dark,
+            button_primary_text_color_dark=primary_dark,
+            color_accent_soft_dark=accent_soft_dark,
+            border_color_accent_subdued_dark=accent_soft_dark,
+
+            block_radius="15px",
+            container_radius="32px",
+        )