additional containers + updated prompts

app.py

@@ -75,7 +75,7 @@ def process_concept(sn_api_key, concept, num_subconcepts, progress=gr.Progress()
     subconcepts = subconcepts[:lmt]
 
     prompts = [
-        f"Please give a detailed explanation of this subconcept: {subconcept}" for subconcept in subconcepts
+        f"Please give a detailed explanation of this subconcept: {subconcept} in the context of this original concept: {concept}" for subconcept in subconcepts
     ]
 
     progress(0.3, f"Generating explanations for {len(subconcepts)} subconcepts in parallel")
@@ -119,11 +119,12 @@ def process_concept(sn_api_key, concept, num_subconcepts, progress=gr.Progress()
 
     progress(0.9, "Formatting output")
     # Format the output in Markdown
+    perf_markdown = f"### Performance\n\n"
+    perf_markdown +=  f"**# of LLama 3.2 3B calls made to SambaNova's API:** {num_subconcepts + 1}\n\n"
+    perf_markdown += f"**# of LLama 3.1 8B calls made to SambaNova's API: 1 (summary)**\n\n"
+    perf_markdown += f"**Total time taken:** {total_time:.2f} seconds\n\n"
+
     markdown_intro = f"# Lesson Plan: {concept.title()}\n\n"
-    markdown_intro += f"**Number of LLama 3.2 3B calls made to SambaNova's API:** {num_subconcepts + 1}\n\n"
-    markdown_intro += f"**1 LLama 3.1 8B call made to SambaNova's API to summarize**\n\n"
-    markdown_intro += f"**Total time taken:** {total_time:.2f} seconds\n\n"
-    markdown_intro += f"\n\n---\n\n"
     markdown_intro += intro_summary
 
     subconcept_markdowns = []
@@ -141,14 +142,14 @@ def process_concept(sn_api_key, concept, num_subconcepts, progress=gr.Progress()
     new_tabs.extend([gr.update(visible=False, render=False) for _ in range(MAX_SUBCONCEPTS-len(subconcept_markdowns))])
     new_tab_contents.extend([gr.update(visible=False) for _ in range(MAX_SUBCONCEPTS-len(subconcept_markdowns))])
 
-    return "Process complete!", markdown_intro, *new_tabs, *new_tab_contents
+    return "Process complete!", perf_markdown, markdown_intro, *new_tabs, *new_tab_contents
 
 with gr.Blocks() as demo:
     gr.Markdown(
         """
         # Lesson Plan Generator
         
-        **Ever wanted to learn something new? Struggled to break down the concept to more digestible subconcepts? In this demo, we use <a href="https://cloud.sambanova.ai">SambaNova's</a> superfast LLama 3.2 3B and LLama 3.1 8B models to summarize the concept and subconcepts as well as provide a detailed lesson for each of the subconcepts.**
+        **Ever wanted to learn something new? Struggled to break down the concept to more digestible subconcepts? In this demo, we use <a href="https://cloud.sambanova.ai">SambaNova's</a> superfast LLama 3.2 3B and LLama 3.1 8B models to expound on the concept and subconcepts, provide a detailed lesson for each of the subconcepts, and summarize the lesson plan.**
         
         *To use this, follow the instructions below:*
 
@@ -173,6 +174,7 @@ with gr.Blocks() as demo:
     
     with gr.Column():
         progress_output = gr.Textbox(label="Progress", interactive=True)
+        performance_output = gr.Markdown(label="Performance", container=True)
         lesson_intro = gr.Markdown(label="Lesson Intro", container=True) 
 
     tab_contents = []
@@ -186,7 +188,7 @@ with gr.Blocks() as demo:
     generate_btn.click(
         process_concept,
         inputs=[sn_api_key_input, concept_input, slider],
-        outputs=[progress_output, lesson_intro] + [tabs.children[i] for i in range(MAX_SUBCONCEPTS)] + tab_contents
+        outputs=[progress_output, performance_output, lesson_intro] + [tabs.children[i] for i in range(MAX_SUBCONCEPTS)] + tab_contents
     )
 
 demo.launch()