Increase input size, use Llama 2, and add SDXL image

app.py

@@ -1,3 +1,4 @@
+import base64
 import os
 import pathlib
 import json5
@@ -281,7 +282,7 @@ def show_bonus_stuff(ppt_headers: List[str]):
     :param ppt_headers: A list of the slide headings.
     """
 
-    # Use the presentation title and the slides headers to find relevant info online
+    # Use the presentation title and the slide headers to find relevant info online
     logging.info('Calling Metaphor search...')
     ppt_text = ' '.join(ppt_headers)
     search_results = get_web_search_results_wrapper(ppt_text)
@@ -294,6 +295,17 @@ def show_bonus_stuff(ppt_headers: List[str]):
         st.markdown('\n\n'.join(md_text_items))
 
     # Avoid image generation. It costs time and an API call, so just limit to the text generation.
+    with st.expander('AI-generated image on the presentation topic'):
+        logging.info('Calling SDXL for image generation...')
+        # img_empty.write('')
+        # img_text.write(APP_TEXT['image_info'])
+        image = get_ai_image_wrapper(ppt_text)
+
+        if len(image) > 0:
+            image = base64.b64decode(image)
+            st.image(image, caption=ppt_text)
+            st.info('Tip: Right-click on the image to save it.', icon="💡️")
+            logging.info('Image added')
 
 
 def main():

global_config.py

@@ -9,9 +9,9 @@ load_dotenv()
 @dataclass(frozen=True)
 class GlobalConfig:
     CLARIFAI_PAT = os.environ.get('CLARIFAI_PAT', '')
-    CLARIFAI_USER_ID = 'meta'
-    CLARIFAI_APP_ID = 'Llama-2'
-    CLARIFAI_MODEL_ID = 'llama2-13b-chat'
+    CLARIFAI_USER_ID = 'clarifai'
+    CLARIFAI_APP_ID = 'ml'
+    CLARIFAI_MODEL_ID = 'llama2-13b-alternative-4k'
 
     CLARIFAI_USER_ID_GPT = 'openai'
     CLARIFAI_APP_ID_GPT = 'chat-completion'
@@ -25,7 +25,7 @@ class GlobalConfig:
     # LLM_MODEL_TEMPERATURE: float = 0.5
     LLM_MODEL_MIN_OUTPUT_LENGTH: int = 50
     LLM_MODEL_MAX_OUTPUT_LENGTH: int = 2000
-    LLM_MODEL_MAX_INPUT_LENGTH: int = 250
+    LLM_MODEL_MAX_INPUT_LENGTH: int = 1500
 
     METAPHOR_API_KEY = os.environ.get('METAPHOR_API_KEY', '')
 

strings.json

@@ -14,7 +14,7 @@
         "Since you have come this far, we have unlocked some more good stuff for you!"
     ],
     "input_labels": [
-        "**Describe the topic of the presentation using 10 to 250 characters. Avoid mentioning the count of slides.**"
+        "**Describe the topic of the presentation using 10 to 1500 characters. Avoid mentioning the count of slides.**"
     ],
     "button_labels": [
         "Generate contents",