Update app.py

app.py

@@ -2,10 +2,12 @@ import gradio as gr
 import requests
 import json
 
+import os
+
 # WARNING: It is not recommended to hardcode sensitive data like API tokens in code.
 # Consider using environment variables or other secure methods for production applications.
 API_URL = "https://deployment.datasaur.ai/api/deployment/8/2717/chat/completions"
-API_TOKEN = "ds-bb8f69148d3e4ae3aed327af3da5eb4a"
+API_TOKEN = os.environ["DATASAUR_API_KEY"]
 
 def magic_function(input_text):
     """
@@ -38,8 +40,11 @@ def magic_function(input_text):
 
 
 with gr.Blocks() as demo:
-    text_area = gr.Textbox(label="Your Text", lines=20)
-    magic_button = gr.Button("Magic Button")
+    gr.Markdown("# Memo Improvement Workflow")
+    with gr.Row():
+        text_area = gr.Textbox(label="Your Text", lines=20, scale=4)
+        with gr.Column(scale=1):
+            magic_button = gr.Button("Magic Button")
 
     magic_button.click(
         fn=magic_function,