Update app.py

app.py

@@ -17,10 +17,6 @@ def generate_text(prompt):
     response = pipe(prompt, max_length=100, num_return_sequences=1)
     return response[0]['generated_text']
 
-# Function to handle example entry selection
-def fill_prompt(selected_prompt):
-    return selected_prompt
-
 # Custom CSS for the app
 css = """
 body {background-color: #f0f8ff; font-family: 'Arial', sans-serif;}
@@ -42,6 +38,10 @@ example_entries = [
     "I learned a new approach to managing my time more effectively today. I’m going to apply this to my daily routine to see if it helps me stay more productive."
 ]
 
+# Function to return the selected prompt
+def fill_prompt(selected_prompt):
+    return gr.update(value=selected_prompt)
+
 # Create the Gradio interface
 with gr.Blocks(css=css) as journal_app:
     
@@ -53,14 +53,12 @@ with gr.Blocks(css=css) as journal_app:
     gr.Markdown("### **Click an example to start writing:**")
     
     with gr.Row():
+        prompt_textbox = gr.Textbox(label="Write your thoughts here:", placeholder="Start writing or select an example...", elem_id="prompt_box", lines=5)
+        
         for i, entry in enumerate(example_entries):
-            gr.Button(f"Example {i+1}").click(fn=lambda e=entry: fill_prompt(e), inputs=[], outputs="prompt_textbox")
+            gr.Button(f"Example {i+1}").click(fn=fill_prompt, inputs=None, outputs=prompt_textbox, _js="")
 
     with gr.Row():
-        with gr.Column():
-            # Input text box with example entries functionality
-            prompt_textbox = gr.Textbox(label="Write your thoughts here:", placeholder="Start writing or select an example...", elem_id="prompt_box", lines=5)
-        
         with gr.Column():
             # Output for generated text
             output = gr.Textbox(label="Your AI Journal Entry ✨", lines=5)
@@ -76,4 +74,4 @@ with gr.Blocks(css=css) as journal_app:
     gr.Markdown("Made with ❤️ using Gradio and Llama-3.2")
 
 # Launch the app
-journal_app.launch()
+journal_app.launch()