Adjust used input part

app.py

@@ -1,10 +1,11 @@
 import random
 import os
 import re
+import streamlit as st
 from langchain_nvidia_ai_endpoints import ChatNVIDIA
 
-# Set environmental variables
-nvidia_api_key = os.getenv("NVIDIA_KEY")
+os.environ["NVIDIA_API_KEY"] = "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"
+nvidia_api_key = os.environ["NVIDIA_API_KEY"]
 
 # Initialize ChatNVIDIA for text generation
 client = ChatNVIDIA(
@@ -81,13 +82,14 @@ def generate(starting_text):
         if response_end != "":
             return response_end
 
-# Example usage
-if __name__ == "__main__":
-    # Get user input for the starting text
-    user_input = input("Enter a starting text for the Midjourney prompt: ")
-
-    # Generate the prompt
-    generated_prompt = generate(user_input)
-
-    # Print the generated prompt
-    print(f"Generated Prompt:\n{generated_prompt}")
+# 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.")