Update app.py

app.py

@@ -568,42 +568,50 @@ def generate(
             output_file = asyncio.run(text_to_speech(final_response, voice))
             yield gr.Audio(output_file, autoplay=True)
 
-# Gradio Chat Interface Setup and Launch
-
-demo = gr.ChatInterface(
-    fn=generate,
-    additional_inputs=[
-        gr.Slider(label="Max new tokens", minimum=1, maximum=MAX_MAX_NEW_TOKENS, step=1, value=DEFAULT_MAX_NEW_TOKENS),
-        gr.Slider(label="Temperature", minimum=0.1, maximum=4.0, step=0.1, value=0.6),
-        gr.Slider(label="Top-p (nucleus sampling)", minimum=0.05, maximum=1.0, step=0.05, value=0.9),
-        gr.Slider(label="Top-k", minimum=1, maximum=1000, step=1, value=50),
-        gr.Slider(label="Repetition penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.2),
-    ],
-    examples=[
-        ["@tts2 What causes rainbows to form?"],
-        ["@3d A birthday cupcake with cherry"],
-        [{"text": "summarize the letter", "files": ["examples/1.png"]}],
-        ["@image Chocolate dripping from a donut against a yellow background, in the style of brocore, hyper-realistic"],
-        ["@rAgent Explain how a binary search algorithm works."],
-        ["@web latest breakthroughs in renewable energy"],
-
-    ],
-    cache_examples=False,
-    type="messages",
-    description=DESCRIPTION,
-    css=css,
-    fill_height=True,
-    textbox=gr.MultimodalTextbox(label="Query Input", file_types=["image"], file_count="multiple"),
-    stop_btn="Stop Generation",
-    multimodal=True,
-)
+# Gradio Chat Interface Setup and Launch using Blocks with Sidebar
+
+with gr.Blocks(css=css) as demo:
+    with gr.Sidebar(label="Usage", open=True):
+        gr.Markdown(
+            "### About Space\n\n"
+            "This chat interface supports multiple commands:\n"
+            "- `@tts1` or `@tts2`: Triggers text-to-speech.\n"
+            "- `@image`: Generates an image using Stable Diffusion XL.\n"
+            "- `@3d`: Generates a 3D model using the ShapE pipeline.\n"
+            "- `@web`: Performs a web search or visits a webpage.\n"
+            "- `@rAgent`: Initiates a reasoning chain using Llama mode OpenAI."
+        )
 
-# Ensure the static folder exists
-if not os.path.exists("static"):
-    os.makedirs("static")
+    chat_interface = gr.ChatInterface(
+        fn=generate,
+        additional_inputs=[
+            gr.Slider(label="Max new tokens", minimum=1, maximum=MAX_MAX_NEW_TOKENS, step=1, value=DEFAULT_MAX_NEW_TOKENS),
+            gr.Slider(label="Temperature", minimum=0.1, maximum=4.0, step=0.1, value=0.6),
+            gr.Slider(label="Top-p (nucleus sampling)", minimum=0.05, maximum=1.0, step=0.05, value=0.9),
+            gr.Slider(label="Top-k", minimum=1, maximum=1000, step=1, value=50),
+            gr.Slider(label="Repetition penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.2),
+        ],
+        examples=[
+            ["@tts2 What causes rainbows to form?"],
+            ["@3d A birthday cupcake with cherry"],
+            [{"text": "summarize the letter", "files": ["examples/1.png"]}],
+            ["@image Chocolate dripping from a donut against a yellow background, in the style of brocore, hyper-realistic"],
+            ["@rAgent Explain how a binary search algorithm works."],
+            ["@web latest breakthroughs in renewable energy"],
+        ],
+        cache_examples=False,
+        type="messages",
+        description=DESCRIPTION,
+        css=css,
+        fill_height=True,
+        textbox=gr.MultimodalTextbox(label="Query Input", file_types=["image"], file_count="multiple"),
+        stop_btn="Stop Generation",
+        multimodal=True,
+    )
 
-from fastapi.staticfiles import StaticFiles
-demo.app.mount("/static", StaticFiles(directory="static"), name="static")
+    # Ensure the static folder is mounted for serving generated files.
+    from fastapi.staticfiles import StaticFiles
+    demo.app.mount("/static", StaticFiles(directory="static"), name="static")
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(share=True)