Update app.py

app.py

@@ -1,13 +1,11 @@
 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 = os.environ["DATASAUR_API_KEY"]
+API_TOKEN = os.getenv("DATASAUR_API_TOKEN")
 
 def magic_function(input_text):
     """
@@ -38,18 +36,29 @@ def magic_function(input_text):
         # Handle cases where response is not valid JSON or structure is unexpected
         return f"Error processing API response: {response.text}"
 
+def handle_magic_click(current_text):
+    """
+    When the magic button is clicked, this function gets the improved text,
+    and returns the new and previous text to update the UI.
+    It also makes the "Previous Text" box visible.
+    """
+    improved_text = magic_function(current_text)
+    return improved_text, current_text, gr.update(visible=True)
+
 
 with gr.Blocks() as demo:
     gr.Markdown("# Memo Improvement Workflow")
     with gr.Row():
-        text_area = gr.Textbox(label="Your Text", lines=20, scale=4)
+        with gr.Column(scale=4):
+            text_area = gr.Textbox(label="Your Text", lines=20)
+            previous_text_area = gr.Textbox(label="Previous Text", lines=20, visible=False)
         with gr.Column(scale=1):
             magic_button = gr.Button("Magic Button")
 
     magic_button.click(
-        fn=magic_function,
+        fn=handle_magic_click,
         inputs=text_area,
-        outputs=text_area
+        outputs=[text_area, previous_text_area, previous_text_area]
     )
 
 if __name__ == "__main__":