Update app.py

app.py

@@ -11,11 +11,25 @@ st.write("This app generates text based on the input prompt using the Gemma mode
 # Text input for user prompt
 prompt = st.text_input("Enter your prompt:", "Once upon a time,")
 
+# User controls for output length and creativity
+max_length = st.slider("Select the maximum output length:", min_value=50, max_value=500, value=100)
+temperature = st.slider("Adjust the creativity level (temperature):", min_value=0.1, max_value=1.0, value=0.7)
+
 # Generate button to trigger text generation
 if st.button("Generate Text"):
-    # Generate text based on the user prompt
-    generated_text = generator(prompt, max_length=100)[0]['generated_text']
-    # Display the generated text
-    st.write("Generated Text:")
-    st.write(generated_text)
+    with st.spinner('Generating text...'):
+        try:
+            generated_text = generator(prompt, max_length=max_length, temperature=temperature)[0]['generated_text']
+        except Exception as e:
+            st.error(f"Error generating text: {str(e)}")
+        else:
+            st.success('Text generation complete!')
+            st.markdown("### Generated Text:")
+            st.markdown(generated_text)
 
+# About section
+with st.expander("About"):
+    st.write("""
+        The Gemma Text Generation app uses the powerful Gemma-7b model from Google to generate text. 
+        Adjust the sliders to change the length and creativity of the output.
+    """)