Upload 23 files

app.py

@@ -59,11 +59,11 @@ context_str = (
     "Instead of saying 'you', you could say something like: 'buddy'."
     "If questions are asked that are not related to fitness, then don't answer them and 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."
-    "Add fitness related emojis to your message."
-    "If someone asks you how to change tires, then don't answer it and play it off cool."
+    "Add fitness related emojis to the end of your message."
+    "Fang deine Sätze immer wieder anders an und grüße den Nutzer nicht. Niemals.."
 )
 
-index = None
+chat_engine = None
 
 def setup_db():
     """
@@ -95,6 +95,8 @@ def setup_db():
 
         # Crop images to tables
         for image in os.listdir("./table_images"):
+            if image.startswith('.DS_'):
+                continue
             detect_and_crop_save_table("./table_images/"+image)
             # Delete old uncropped image
             os.remove("./table_images/"+image)
@@ -129,7 +131,7 @@ def setup_ai():
     assigns the context_template and system_prompt used for manipulating
     the AI responses.
     """
-    global openai_mm_llm, context_str, system_prompt, index
+    global openai_mm_llm, context_str, system_prompt, chat_engine
 
     # Setup database
     text_documents, image_documents, storage_context = setup_db()
@@ -147,20 +149,22 @@ def setup_ai():
         storage_context=storage_context
     )
 
+    # Create a chat engine from the index
+    chat_engine = index.as_chat_engine(
+        chat_mode="context",
+        system_prompt=system_prompt,
+        context_template=context_str
+    )
+
 def response(message, history):
     """
     Get a reponse from OpenAI and send the chat_history with every query.
     """
-    global index
+    global chat_engine
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
 
-    # Create a chat engine from the index
-    chat_engine = index.as_chat_engine(
-        chat_mode="context",
-        system_prompt=system_prompt,
-        context_template=context_str
-    )
+    
     # Re-use chat_history & sanity check
     # We do this because the chat_engine expects a list
     # of some sort when using chat_history.
@@ -196,12 +200,12 @@ def main():
         avatar_images=("user_avatar.png", "chatbot_avatar.png"),
         layout='bubble',
         show_label=False,
-        height=400,
+        height=500,
         value=[[None, "Griaß di, i bin da Arnold, dei Fitness-Chatbot. I gfrei mi, dass du di für dei Gsundheit und dein Leib interessierst. I kann da helfen, deine Fitnessziele zu erreichen, indem i da Fragen zu Muskeln, Training, BMI und vü mehr beantworte. Du kannst gern mit mir experimentieren und herausfinden, was i sunst no so drauf hob. I bin immer bereit, da zu helfen. Wos wüst wissen? 💪🏋️💬"]]
     )
 
     submit_button = gradio.Button(
-        value="Ask Arnold",
+        value="ASK ARNOLD",
         elem_classes=["ask-button"],
     )
 

style.css

@@ -22,13 +22,17 @@ div.gap.panel>div.gr-group {
 }
 
 .ask-button {
-    background-color: var(--color-accent);
+    /*background-color: var(--color-accent);*/
+    background-color: #5fb0ab !important;
     font-weight: bold;
     letter-spacing: 0.1rem;
 }
 
 .gallery-item {
-    color: var(--block-label-text-color) !important
+    color: var(--block-label-text-color) !important;
+    border: 1px var(--block-border-color) solid !important;
+    padding: 2px !important;
+    margin: 2px !important;
 }
 
 div.message-wrap {
@@ -41,6 +45,6 @@ div.message-wrap {
     text-align: center;
 }
 
-.generating {
-    border: 0px solid transparent;
+div.generating {
+    border: none !important;
 }