Extra buttons

app.py

@@ -77,10 +77,25 @@ def custom_generate_response2(cust_user_message):
     cust_user_message = CustomPrompts[2] + '\n' + cust_user_message + '\n\nScene Details'
     yield from generate_response(cust_user_message)
 
+def custom_generate_response3(cust_user_message):
+    cust_user_message = CustomPrompts[3] + '\n' + cust_user_message + '\n\nTeardown Details:'
+    yield from generate_response(cust_user_message)
+
+def custom_generate_response4(cust_user_message):
+    cust_user_message = CustomPrompts[4] + '\n' + cust_user_message + '\n\nManufacturing Details:'
+    yield from generate_response(cust_user_message)
+
+def custom_generate_response5(cust_user_message):
+    cust_user_message = CustomPrompts[5] + '\n' + cust_user_message + '\n\nConsiderations:'
+    yield from generate_response(cust_user_message)
+
 CustomPrompts = [
     "Write a Class Diagram based on the following text:",
     "Write a Pydot code based on the following text:",
     "Describe what a standard happy scene in any movie would be planned in great detail, based on the following text:",
+    "Explain a teardown of the product mentioned in the following text:",
+    "Explain the manufacturing of the product mentioned in the following text:",
+    "Explain the marketing considerations of the product mentioned in the following text:",
 ]
 
 with gr.Blocks() as iface: 
@@ -88,7 +103,7 @@ with gr.Blocks() as iface:
         fn=generate_response,
         inputs=gr.Textbox(lines=2, placeholder="Type your message here..."),
         outputs="text",
-        title="Stable LM 2 Zephyr (1.6b) LLama.cpp Interface Test",
+        title="Stable LM 2 Zephyr (1.6b) LLama.cpp Interface Test (Inconsistent Performance - 100 tokens in 50 secs or 800+ secs)",
         description="No Prompt template used yet (Essentially autocomplete). No Message History for now - Enter your message and get a response.",
         flagging_dir="/usr/src/app/flagged",
     )
@@ -97,11 +112,16 @@ with gr.Blocks() as iface:
         gr.HTML("Test for wrapping generator (Instead of buttons tabs and dropdowns?)")
         MainOutput = gr.TextArea(placeholder='Output will show here')
         CustomButtonInput = gr.TextArea(lines=1, placeholder='Prompt goes here')
+        CustomButtonClassDiagram = gr.Button(CustomPrompts[3])
+        CustomButtonClassDiagram = gr.Button(CustomPrompts[4])
+        CustomButtonClassDiagram = gr.Button(CustomPrompts[5])
         CustomButtonClassDiagram = gr.Button(CustomPrompts[0])
         CustomButtonPydotcode = gr.Button(CustomPrompts[1])
         CustomButtonHappyMovieScene = gr.Button(CustomPrompts[2])
         CustomButtonClassDiagram.click(custom_generate_response0, inputs=[CustomButtonInput], outputs=MainOutput)
         CustomButtonPydotcode.click(custom_generate_response1, inputs=[CustomButtonInput], outputs=MainOutput)
         CustomButtonHappyMovieScene.click(custom_generate_response2, inputs=[CustomButtonInput], outputs=MainOutput)
+        CustomButtonHappyMovieScene.click(custom_generate_response3, inputs=[CustomButtonInput], outputs=MainOutput)
+        CustomButtonHappyMovieScene.click(custom_generate_response4, inputs=[CustomButtonInput], outputs=MainOutput)
 
 iface.queue().launch(server_name="0.0.0.0", share=True)