update

app.py

@@ -34,7 +34,6 @@ def main():
         #except:
         #    pass
 
-
         llm_settings = config['llm']
         config['llm']['api_key'] = os.environ["OPENAI_KEY"]
         newsletter_meta_info = config['newsletter']
@@ -54,6 +53,8 @@ def main():
             max_tokens,
             system_message, 
             textual_preferences,
+            few_shot=None,
+            custom_template=None,
             progress=gr.Progress()
             ):
 
@@ -67,12 +68,15 @@ def main():
             max_transactions=newsletter_meta_info['max_recents_items']) 
         logging.debug(f"Recommendations: {recommendations}")
         logging.debug(f"Transactions: {transactions}")
-        print("cusomter info", customer_info)
+        print("customer info", customer_info)
 
 
         # Load the html template and replace the placeholders for images with the actual content
         logging.info("Initializing newsletter template...")
         progress(0.5, "Initializing personalized content...")
+        # override the default template if a custom one is provided
+        if custom_template:
+            newsletter_meta_info['newsletter_example_path'] = custom_template
         newsletter_text = initialize_newsletter(newsletter_meta_info, transactions, recommendations)
 
         # Build context from the user preferences, the recommendations and the transactions
@@ -85,7 +89,7 @@ def main():
 
         # Build the prompt for the LLM
         progress(0.7, "Generating personalized content...")
-        prompt = build_prompt(context)
+        prompt = build_prompt(context, few_shot)
         logging.info(f"Prompt: {prompt}")
         
 
@@ -156,12 +160,28 @@ def main():
                     value=llm_handler.default_max_tokens, 
                     precision=0
                 )
+
+                custom_template = gr.File(
+                    label="Custom Template", 
+                    file_types="html", 
+                    visible=True)
+
+            with gr.Row():
                 system_message = gr.Textbox(
                     label="System Message", 
                     placeholder="Enter a custom system message (optional).", 
                     value=llm_handler.default_system_message,
                     visible=False
                 )
+                
+                few_shot = gr.Textbox(
+                    label="Few Shot", 
+                    placeholder="Enter a custom few shot (optional).", 
+                    value=None,
+                    visible=True)
+                
+                
+
         
         # User Context (Hidden by Default)
         with gr.Accordion("🧑‍💻 User Context", open=False, visible=False):
@@ -188,7 +208,9 @@ def main():
                 temperature,
                 max_tokens,
                 system_message,
-                textual_preferences
+                textual_preferences,
+                few_shot,
+                custom_template
             ], 
             outputs=[newsletter_output, download]
         )
@@ -200,48 +222,5 @@ def main():
     )
 
 
-    # Gradio interface for the app
-"""    logging.info("Creating interface...")
-    with gr.Blocks() as demo:
-        gr.Markdown("### Newsletter Generator")
-        
-        customer_id = gr.Textbox(label="Client ID", value="04a183a27a6877e560e1025216d0a3b40d88668c68366da17edfb18ed89c574c")
-        textual_preferences = gr.Textbox(label="Newsletter preferences", placeholder="The newsletter should be catchy.")
-        
-        # llm_preferences = gr.Textbox(label="LLM Preferences", placeholder="Enter LLM preferences.", visible=False)
-        # create an openable block for the llm preferences
-        with gr.Accordion("LLM Preferences", open=False):
-            model_name = gr.Dropdown(label="Model Name", choices=["gpt-3.5-turbo", "gpt-4o"], value=llm_handler.model_name)
-            temperature = gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, step=0.05, value=llm_handler.default_temperature)
-            max_tokens = gr.Number(label="Max Tokens", value=llm_handler.default_max_tokens)
-            system_message = gr.Textbox(label="System Message", placeholder="Enter the system message or Leave Blank.", value=llm_handler.default_system_message)
-        
-        with gr.Accordion("User Context", open=False, visible=False):
-            # get profiled user context
-            pass
-        
-        generate_button = gr.Button("Generate Newsletter")
-        
-        # create a button to open the  newsletter in a new tab
-        download = gr.DownloadButton(label="Download Newsletter")
- 
-        newsletter_output = gr.HTML(label="Generated Newsletter", min_height="500", value="<br><br><br><br><br>")
-        
-        generate_button.click(
-            fn=generate_newsletter, 
-            inputs=[
-                customer_id,
-                # llm preferences
-                model_name,
-                temperature,
-                max_tokens,
-                system_message, 
-                # newsletter preferences
-                textual_preferences], 
-            outputs=[newsletter_output, download]
-        )
-    
-    demo.queue().launch(share=config['app']['share'], server_port=config['app']['server_port'])"""
-
 if __name__ == "__main__":
     main()

config/config.yaml

@@ -19,6 +19,7 @@ llm:
             "recommendations": "your recommendations text here",
             "closing": "your closing text here"
         }
+default_few_shot: null
 
 
 recommender_api:
@@ -34,7 +35,7 @@ app:
 newsletter:
   newsletter_example_path: "./newsletter_examples/1.html"
   brand_logo: "https://seeklogo.com/images/L/luisa-spagnoli-logo-EF482BEE89-seeklogo.com.png"
-  brand_name: "AI x Fashion"
+  brand_name: "Luisa Spagnoli"
   max_recommendations: 2
   max_recents_items: 2
 

src/utils.py

@@ -52,7 +52,9 @@ def format_personalization(personalization):
 
 
 def build_prompt(
-        context: Dict[str, Any]) -> str:
+        context: Dict[str, Any],
+        few_shot: str = None,
+        ) -> str:
         """
         Create a detailed prompt incorporating all personalization factors
         
@@ -62,6 +64,11 @@ def build_prompt(
         :return: Formatted prompt string
         """
 
+        if few_shot is not None and len(few_shot) > 0:
+            few_shot = f"Few-shot examples:\n{few_shot}\n"
+        else:
+            few_shot = ""
+
         # Create a detailed user prompt with all context and personalization
         user_prompt = f"""
         Generate personalized newsletter content based on the following information:
@@ -79,13 +86,14 @@ def build_prompt(
         3. recommendations: Present recommended items naturally, explaining why they match the customer's preferences and previous purchases.
         4. closing: Wrap up maintaining the established tone. 
 
-        General reequirements:
+        General requirements:
         - Write in first person. 
         - Don't be too formal, maintain a friendly, warm and engaging tone, avoid stuff like "I hope you are doing well".
         - The reader should feel like they have a special dedicated fashion assistant, who is also a friend.
         - When mentioning items, use the item name and avoid mentioning colors that are not present in the image, or using adjectives like "vibrant".
         - The goal should be to make the reader feel special and excited about the items they are being recommended.
 
+        {few_shot}
         
         Please generate all sections in a single, coherent response that maintains consistency in tone and style throughout. Begin each section with the section name in all caps.
         """