Update pages/descriptionGen.py

pages/descriptionGen.py

@@ -13,7 +13,7 @@ prompt_file = "prompt_template.txt"
 
 
 class ProductDescGen(LLMChain):
-    """LLM Chain specifically for generating multi paragraph rich text product description using emojis."""
+    """LLM Chain specifically for generating multi-paragraph rich text product description using emojis."""
 
     @classmethod
     def from_llm(
@@ -23,7 +23,7 @@ class ProductDescGen(LLMChain):
         return cls(llm=llm, prompt=prompt, **kwargs)
 
 
-def product_desc_generator(product_name, keywords, OPENAI_API):
+def product_desc_generator(product_name, keywords, openai_api_key):
     with open(prompt_file, "r") as file:
         prompt_template = file.read()
 
@@ -34,7 +34,7 @@ def product_desc_generator(product_name, keywords, OPENAI_API):
         model_name="gpt-3.5-turbo",
         temperature=0.7,
         # openai_api_key=OPENAI_API_KEY,
-        openai_api_key=OPENAI_API,
+        openai_api_key=openai_api_key,
     )
 
     ProductDescGen_chain = ProductDescGen.from_llm(llm=llm, prompt=PROMPT)
@@ -49,17 +49,20 @@ def main():
         "Generate multi-paragraph rich text product descriptions for your products instantly!"
         " Provide the product name and keywords related to the product."
     )
-    OPENAI_API = st.text_input("Open Ai Key", "yut_86876vv87tbyuttvubyurvt")
+    openai_api_key = st.text_input("OpenAI API Key", "your_openai_api_key_here")
     product_name = st.text_input("Product Name", "Nike Shoes")
-    keywords = st.text_input("Keywords (separated by commas)", "black shoes, leather shoes for men, water resistant")
+    keywords = st.text_input(
+        "Keywords (separated by commas)",
+        "black shoes, leather shoes for men, water resistant"
+    )
 
     if st.button("Generate Description"):
-        OPENAI_API:
-            description = product_desc_generator(product_name, keywords, OPENAI_API)
+        if openai_api_key:
+            description = product_desc_generator(product_name, keywords, openai_api_key)
             st.subheader("Product Description:")
             st.text(description)
         else:
             st.warning("Please provide your OpenAI API Key.")
 
 if __name__ == "__main__":
-    main()
+    main()