Update app.py

app.py

@@ -9,13 +9,15 @@ import os
 # Set up page configuration
 st.set_page_config(page_title="ProductProse - AI Product Description Generator", layout="wide")
 
-# Initialize session state to track API responses and query count
+# Initialize session state to track API responses and user feedback
 if 'generated_description' not in st.session_state:
     st.session_state.generated_description = None
 if 'translated_description' not in st.session_state:
     st.session_state.translated_description = None
 if 'customized_description' not in st.session_state:
     st.session_state.customized_description = None
+if 'feedback' not in st.session_state:
+    st.session_state.feedback = None
 
 # Sidebar for product data input
 st.sidebar.title("Product Data Input")
@@ -25,7 +27,7 @@ benefits = st.sidebar.text_area("Product Benefits", "Benefit 1, Benefit 2, Benef
 specifications = st.sidebar.text_area("Product Specifications", "Specification 1, Specification 2, Specification 3")
 
 # Select target language for translation
-target_language = st.sidebar.selectbox("Target Language for Translation", ["French", "Spanish", "German", "Chinese", "Japanese"])
+target_language = st.sidebar.selectbox("Target Language for Translation", ["Arabic", "Chinese", "French", "German", "Japanese", "Portugese", "Russian", "Spanish", "Urdu"])
 
 # Main app title and description
 st.title("ProductProse - AI Product Description Generator")
@@ -43,7 +45,13 @@ if api_key and project_id:
     client = APIClient(credentials)
     client.set.default_project(project_id)
 
-    # Generate Product Description
+    # Tone Selection for Description Customization
+    tone = st.sidebar.selectbox("Select Tone for Description", ["Formal", "Casual", "Professional", "Playful"])
+
+    # Keyword Input for SEO Optimization
+    seo_keywords = st.sidebar.text_area("SEO Keywords (comma-separated)", "smart home, intelligent, automation")
+
+    # Step 1: Generate Product Description
     st.header("Step 1: Generate Product Description")
     if st.button("Generate Description"):
         if product_name and features and benefits and specifications:
@@ -74,7 +82,7 @@ if api_key and project_id:
         else:
             st.warning("Please fill in all the product data fields before generating a description.")
 
-    # Translate Product Description
+    # Step 2: Translate Product Description
     st.header("Step 2: Translate Product Description")
     if st.session_state.generated_description:
         if st.button("Translate Description"):
@@ -105,15 +113,15 @@ if api_key and project_id:
         st.subheader(f"Translated Product Description ({target_language})")
         st.write(st.session_state.translated_description)
 
-    # Customize Product Description via Chat Interface
+    # Step 3: Customize Product Description via Chat Interface
     st.header("Step 3: Customize Product Description")
-    customization_prompt = st.text_input("Customize the product description (e.g., adjust tone, add brand-specific details)")
+    customization_prompt = st.text_input(f"Customize the product description with a {tone} tone and the following SEO keywords: {seo_keywords}")
 
     if st.session_state.generated_description and customization_prompt:
         if st.button("Customize Description"):
             try:
                 # Customize the description using Granite-13B-Chat
-                prompt = f"Customize the following product description based on the user's request:\n{st.session_state.generated_description}\nCustomization Request: {customization_prompt}\nGenerate only the final customized product description."
+                prompt = f"Customize the following product description with a {tone} tone, using the following SEO keywords: {seo_keywords}.\nProduct Description:\n{st.session_state.generated_description}\nCustomization Request: {customization_prompt}\nGenerate only the final customized product description."
                 model = ModelInference(model_id=ModelTypes.GRANITE_13B_CHAT_V2, params={
                     GenParams.DECODING_METHOD: DecodingMethods.GREEDY,
                     GenParams.MIN_NEW_TOKENS: 50,
@@ -134,5 +142,15 @@ if api_key and project_id:
         st.subheader("Customized Product Description")
         st.write(st.session_state.customized_description)
 
+    # Step 4: Feedback and Quality Scoring
+    st.header("Step 4: Provide Feedback")
+    feedback = st.slider("Rate the quality of the generated product description (1 = Poor, 5 = Excellent)", 1, 5, 3)
+    feedback_comments = st.text_area("Additional Comments")
+
+    if st.button("Submit Feedback"):
+        st.session_state.feedback = {"rating": feedback, "comments": feedback_comments}
+        st.success("Thank you for your feedback!")
+        st.write(st.session_state.feedback)
+
 else:
     st.error("IBM WatsonX API credentials are not set. Please check your environment variables.")