Update app.py

app.py

@@ -44,7 +44,7 @@ def get_chatgpt_response(messages, selected_model):
     return response['choices'][0]['message']['content']
 
 # Define pages in sidebar
-page = st.sidebar.radio('Select a page:', ('Review Analysis', 'review summary', 'Feature Benefits', 'Identify Avatars', 'Tone of Voice Manual', 'question_answers'))
+page = st.sidebar.radio('Select a page:', ('Review Analysis', 'review summary', 'Feature Benefits', 'feature_mapping', 'benefit_mapping', 'Identify Avatars', 'Tone of Voice Manual', 'question_answers'))
 
 
 if page == 'Review Analysis':
@@ -1182,3 +1182,249 @@ if page == 'question_answers':
             st.error(f"An error occurred while writing to Google Sheets: {e}")
 
 
+if page == 'feature_mapping':
+    st.title('feature_mapping')
+
+    gc = gspread.service_account_from_dict(credentials_dict)
+
+    # Ask user for Google Sheet URL
+    sheet_url = st.text_input('Enter the URL of your Google Sheet')
+
+    # Add this in the part of your Streamlit app where the user can select the model
+    models = ["gpt-3.5-turbo", "gpt-4"]
+    selected_model = st.selectbox("Choose a model:", models)
+
+    # Extract sheet ID from URL
+    sheet_id = sheet_url.split('/')[5]
+    # Open the Google spreadsheet
+    sheet = gc.open_by_key(sheet_id)
+
+    # Select sheet named 'Listing'
+    worksheet = sheet.worksheet('Listing')
+    # Get all records of the data
+    df = get_as_dataframe(worksheet)
+
+    # Convert DataFrame to strings
+    df_str = df.astype(str)
+
+    title = df[df.eq('Title').any(axis=1)].iloc[0, 2]
+    bullets = [df[df.eq(f"Bullet #{i}").any(axis=1)].iloc[0, 2] for i in range(1, 6)]
+
+    backend_search_terms = df[df.astype(str).apply(lambda x: 'Backend Search Terms' in ' '.join(x), axis=1)].iloc[0, 2] if len(df[df.astype(str).apply(lambda x: 'Backend Search Terms' in ' '.join(x), axis=1)]) > 0 else None
+
+    image_text_row = df[df.eq('Image Text').any(axis=1)]
+    image_text = list(image_text_row.dropna(axis=1).iloc[0, :])
+
+    a_plus_desc_mask = df.astype(str).apply(lambda x: 'A+ Description' in ' '.join(x), axis=1)
+    if a_plus_desc_mask.any():
+        a_plus_desc_row = df[a_plus_desc_mask].index[0]
+        a_plus_desc = df.iloc[a_plus_desc_row:, :].fillna('').values.flatten()
+        a_plus_desc = ' '.join(a_plus_desc).strip()
+    else:
+        a_plus_desc = None
+
+    legacy_desc = df[df.astype(str).apply(lambda x: 'Legacy Product Description' in ' '.join(x), axis=1)].iloc[0, 2] if len(df[df.astype(str).apply(lambda x: 'Legacy Product Description' in ' '.join(x), axis=1)]) > 0 else None
+
+    image_text_string = ' '.join(image_text)
+
+    bullet_str = ""
+    for i, bullet in enumerate(bullets, 1):
+        bullet_str += f"Bullet #{i}: {bullet}\n"
+
+    # Display default prompt to user
+    default_prompt = """Features: 
+1) Professional Artist Testing: Developed and tested by professional artists to ensure exceptional product quality and results for all types of artistic expression.
+2) 1:1 Aspect Ratio: Square aspect ratio of 1:1 facilitates an easy design process and balanced compositions based on famous artworks.
+3) 100% Cotton Fibers: Uses 100% cotton fibers to provide natural and lasting beauty and exceptional resistance to scratching and erasing.
+4) Gelatin-sized Paper: Offers astounding absorbency resulting in harmonious, natural grain texture and no optical brightening agents that maintain the original brightness of the artwork.
+5) Cold & Hot Presses: Available in both cold press and hot press finishes, resulting in slightly textured, toothed surfaces (cold press) for depth and almost no tooth (hot press) surfaces for seamless blending, overlay, or paint removal.
+6) European Mill Crafting: Crafted by European mill masters with over 400 years of paper-making experience, ensuring that you produce your finest works.
+7) Easy-Block Format: Offers a unique "Easy-Block" format that eliminates the need to stretch watercolor paper sheets, saves time and effort, and ensures that paint dries to the desired texture and prevent buckling or creasing.
+8) 2 Glued, 2 Open Edges: Glued on 2 edges and open on 2 edges for easy removal of finished painting, saving time and effort while maintaining paper integrity and preservation.
+9) Travel-friendly: Suitable for travel, providing increased convenience to artists.
+10) Various Paper Weights: Available in a range of paper weights, ensuring maximum versatility for different artistic styles and techniques.
+11) 100 Series Youth Quality: Offers youth quality in the 100 series, providing a traditional watercolor feel for artistic expression.
+12) Cotton & Wood Pulp Choices: Offers two different choices of watercolor paper: 100% cotton pulp and 100% wood pulp.
+13) Indoor & Outdoor Use: Designed for use indoors and outdoors, providing versatility for artists on the go.
+14) 14 Cold-pressed Sheets: Offers 14 sheets of cold-pressed, textured watercolor paper, providing ample surface area for larger projects.
+15) Glue Binding: Offers glue binding for easy tearing out of pages, allowing for experimentation and perfecting techniques.
+16) Drawing, Sketching, Painting: Versatile for drawing, sketching, and painting.
+17) Mixed Media Suitable: Top choice for mixed media projects.
+18) Artist-approved: Tested and approved by artists.
+19) Chlorine-free & pH-neutral: Chlorine-free, pH-neutral, and archival, ensuring the longevity of artwork.
+20) Natural White Paper: Natural white, providing a textured surface for extraordinary vibrant colors.
+21) Rough Watercolor Paper: Rough, providing both smooth and textured surfaces for experimentation.
+22) Dual-Sided Surfaces: Dual-sided with smooth and textured surfaces.
+23) Expert Quality: Provides expert quality for perfect combination of durability and ease of use.
+24) Art-N-Fly Compatibility: Compatible with Art-N-Fly Watercolor Pens.
+25) Historic Paper Quality: Offers exceptional papers since the 1700s.
+26) Calcium Carbonate Buffer: Buffered with calcium carbonate for added protection.
+27) Random Texture: Provides a distinctive random texture.
+28) Internally Sized: Internally sized to maintain paper integrity and prevent paper deterioration.
+
+
+--------------------------------------------------------------------------------------------
+
+You have to search for presence of above features in below information of listing:
+
+--------------------------------------------------------------------------------------------
+    ------------------------------------
+    Title: {}\n 
+    bullet_str: {}\n
+    Legacy Product Description: {}\n
+    A+ Description: {}\n
+    Image_text: {}\n
+    backend_search_terms: {}\n
+    ------------------------------------- 
+
+
+
+    Please carefully read the given information of listing and create feature present matrix. if feature is present then you have to mark it using 1 and if absent leave empty. please create in table format. note, only present features should be covered. You have to explain at the end where did you find the present features. also once table is ready create one json that I can easily load in pandas df
+            """.format(title, bullet_str, legacy_desc, a_plus_desc, image_text_string, backend_search_terms)
+
+    prompt = st.text_area("Edit the prompt:", value=default_prompt, height=200)
+
+
+
+    # If the user has input a URL, fetch the data from Google Sheets
+    if st.button('generate'):
+
+        messages = [
+                    {"role": "system", "content": "You are helpful assistant"},
+                    {"role": "user", "content": prompt}
+                ]
+
+        
+
+        model_response = get_chatgpt_response(messages, selected_model)
+
+        feature_mapping= model_response
+
+        st.session_state.feature_mapping = feature_mapping
+
+        st.write(feature_mapping)
+
+
+if page == 'benefit_mapping':
+    st.title('benefit_mapping')
+
+    
+    gc = gspread.service_account_from_dict(credentials_dict)
+
+    # Ask user for Google Sheet URL
+    sheet_url = st.text_input('Enter the URL of your Google Sheet')
+
+    # Add this in the part of your Streamlit app where the user can select the model
+    models = ["gpt-3.5-turbo", "gpt-4"]
+    selected_model = st.selectbox("Choose a model:", models)
+
+    # Extract sheet ID from URL
+    sheet_id = sheet_url.split('/')[5]
+    # Open the Google spreadsheet
+    sheet = gc.open_by_key(sheet_id)
+
+    # Select sheet named 'Listing'
+    worksheet = sheet.worksheet('Listing')
+    # Get all records of the data
+    df = get_as_dataframe(worksheet)
+
+    # Convert DataFrame to strings
+    df_str = df.astype(str)
+
+    title = df[df.eq('Title').any(axis=1)].iloc[0, 2]
+    bullets = [df[df.eq(f"Bullet #{i}").any(axis=1)].iloc[0, 2] for i in range(1, 6)]
+
+    backend_search_terms = df[df.astype(str).apply(lambda x: 'Backend Search Terms' in ' '.join(x), axis=1)].iloc[0, 2] if len(df[df.astype(str).apply(lambda x: 'Backend Search Terms' in ' '.join(x), axis=1)]) > 0 else None
+
+    image_text_row = df[df.eq('Image Text').any(axis=1)]
+    image_text = list(image_text_row.dropna(axis=1).iloc[0, :])
+
+    a_plus_desc_mask = df.astype(str).apply(lambda x: 'A+ Description' in ' '.join(x), axis=1)
+    if a_plus_desc_mask.any():
+        a_plus_desc_row = df[a_plus_desc_mask].index[0]
+        a_plus_desc = df.iloc[a_plus_desc_row:, :].fillna('').values.flatten()
+        a_plus_desc = ' '.join(a_plus_desc).strip()
+    else:
+        a_plus_desc = None
+
+    legacy_desc = df[df.astype(str).apply(lambda x: 'Legacy Product Description' in ' '.join(x), axis=1)].iloc[0, 2] if len(df[df.astype(str).apply(lambda x: 'Legacy Product Description' in ' '.join(x), axis=1)]) > 0 else None
+
+    image_text_string = ' '.join(image_text)
+
+    bullet_str = ""
+    for i, bullet in enumerate(bullets, 1):
+        bullet_str += f"Bullet #{i}: {bullet}\n"
+
+    # Display default prompt to user
+    default_prompt = """Benefits:
+1) Long-lasting Artwork: Exceptional product quality ensures long-lasting, beautiful artwork and exceptional results for various artistic expressions.
+2) Easy Composition: A square aspect ratio of 1:1 on the paper facilitates an easy design process and balanced compositions based on famous artworks.
+3) Natural Beauty: 100% cotton fibers provide natural and lasting beauty and exceptional resistance to scratching and erasing.
+4) Superior Absorbency: Gelatin-sized paper offers astounding absorbency resulting in harmonious, natural grain texture and no optical brightening agents that maintain the original brightness of the artwork.
+5) Versatile Finishes: Available in both cold press and hot press finishes result in slightly textured, toothed surfaces (cold press) for depth and almost no tooth (hot press) surfaces for seamless blending, overlay, or paint removal and provide vibrantly versatile paper for customization based on the artist's preference.
+6) Expert Craftsmanship: Crafted by experienced mill masters guarantees that artists produce their finest works.
+7) No Stretching Needed: Unique "Easy-Block" format eliminates the need to stretch watercolor paper sheets, saves time and effort, and ensures that paint dries to the desired texture and prevents buckling or creasing.
+8) Easy Painting Removal: Glued on 2 edges and open on 2 edges for easy removal of finished painting saves time and effort while maintaining paper integrity and preservation.
+9) Travel Convenience: Suitable for travel, providing increased convenience to artists.
+10) Versatility: Available in a range of paper weights ensures maximum versatility for different artistic styles and techniques.
+11) Time & Money Savings: Saves time and money with ready-made sizes.
+12) Durable Preservation: Durable material for long-lasting preservation.
+13) Traditional Feel: Traditional watercolor feel for artistic expression.
+14) Ample Surface Area: Provides ample surface area for larger projects.
+15) Superior Value: Superior value compared to other brands.
+16) Artistic Versatility: Approved by professional artists for various artistic endeavors.
+17) Mixed Media Ideal: Suitable for creating mixed media artwork.
+18) Professional Quality: Ensures the longevity of artwork with high-quality material.
+19) Artwork Longevity: Provides a textured surface for extraordinary vibrant colors, contributing to artwork longevity.
+20) Beautiful Paintings: Ideal for creating beautiful artwork for weddings, gifts, and more.
+21) Vibrant Colors: Provides a textured surface that results in extraordinary vibrant colors.
+22) Technique Experimentation: Offers both smooth and textured surfaces for experimentation with different techniques.
+23) Creative Possibilities: Versatile enough for painting, sketching, printmaking, or drawing.
+24) Durable & Easy: Smooth edges and easy tear sheets ensure no damage to artwork.
+25) Professional Look: Provides a professional appearance for watercolor paintings.
+26) Unique Artisan Touch: Innovative artisan paper adds a unique touch to artwork.
+27) Added Protection: High-quality material ensures artwork longevity and protection.
+28) Budget-friendly: Cost-effective paper for budget-conscious artists.
+29) Quality Assurance: Highest grade of watercolor paper available, providing quality assurance and contributing to artistic confidence.
+
+
+--------------------------------------------------------------------------------------------
+
+You have to search for presence of above Benefits in below information of listing:
+
+--------------------------------------------------------------------------------------------
+    ------------------------------------
+    Title: {}\n 
+    bullet_str: {}\n
+    Legacy Product Description: {}\n
+    A+ Description: {}\n
+    Image_text: {}\n
+    backend_search_terms: {}\n
+    ------------------------------------- 
+
+
+
+    Please carefully read the given information of listing and create Benefits present matrix. if Benefit is present then you have to mark it using 1 and if absent leave empty. please create in table format. note, only present Benefits should be covered. please include benefit name in the table. You have to explain at the end where did you find the present Benefits. also once table is ready create one json that I can easily load in pandas df
+            """.format(title, bullet_str, legacy_desc, a_plus_desc, image_text_string, backend_search_terms)
+
+    prompt = st.text_area("Edit the prompt:", value=default_prompt, height=200)
+
+
+
+    # If the user has input a URL, fetch the data from Google Sheets
+    if st.button('generate'):
+
+        messages = [
+                    {"role": "system", "content": "You are helpful assistant"},
+                    {"role": "user", "content": prompt}
+                ]
+
+        
+
+        model_response = get_chatgpt_response(messages, selected_model)
+
+        benefit_mapping= model_response
+
+        st.session_state.feature_mapping = benefit_mapping
+
+        st.write(benefit_mapping)