Adjust the streamlit input part

app.py

@@ -4,8 +4,8 @@ import re
 import streamlit as st
 from langchain_nvidia_ai_endpoints import ChatNVIDIA
 
-os.environ["NVIDIA_API_KEY"] = "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"
-nvidia_api_key = os.environ["NVIDIA_API_KEY"]
+# Set environmental variables
+nvidia_api_key = os.getenv("NVIDIA_KEY")
 
 # Initialize ChatNVIDIA for text generation
 client = ChatNVIDIA(
@@ -83,13 +83,11 @@ def generate(starting_text):
             return response_end
 
 # Streamlit UI
-st.title("Generate MidJourney Prompts")
-user_input = st.text_input("Enter a starting text for the MidJourney prompt:")
-
-if st.button("Generate"):
-    if user_input:
-        generated_prompt = generate(user_input)
-        st.write("**Generated Prompt:**")
-        st.write(generated_prompt)
-    else:
-        st.warning("Please enter a starting text.")
+st.title("Generate Sci-fi/Space-themed MidJourney Prompts")
+
+user_input = st.text_input("Enter a starting text or just press Generate to get a space-themed prompt:")
+
+if st.button("Generate Prompt"):
+    generated_prompt = generate(user_input)
+    st.write("**MidJourney Prompt:**")
+    st.write(generated_prompt)