Spaces:

VinucaAI
/

Vinuca_AI

Runtime error

Ethan Sue

cleaned up the AI assistant UI

b23a502 almost 2 years ago

3.98 kB

"""
import gradio as gr
def greet(name):
    return "Hello " + name + "!!"
demo = gr.Interface(fn=greet, inputs="text", outputs="text")
demo.launch()"""

import gradio as gr
import pandas as pd
import torch
from sentence_transformers import SentenceTransformer, util
import google.generativeai as genai
import pathlib
import textwrap
from IPython.display import display
from IPython.display import Markdown
import time

# 1. Initialization - Load Data and Set Up model
url = 'https://raw.githubusercontent.com/TRASJEPS/Vinicunca_AI_Project1/refs/heads/main/_Data-20241116T221029Z-001/Data/Cleaned%20Data/Combined%20Cleaned%20Data.csv'
df = pd.read_csv(url)

# Combine relevant columns
df["combined"] = (
    "Product Name: " + df.Product_Name.str.strip()+"; Brand: " + df.Brand.str.strip() 
    + "; Category: " + df.Product_Category.str.strip()
    + "; Details: " + df.Product_Details.str.strip()
    + "; Ingredients: " + df.Ingredients.str.strip()
    + "; Price: " + df["Cleaned Price"].str.strip()
    # +"; desc: "+ df.text.str.strip()
)

# Ensure the 'combined' column has no NaN or invalid values
df['combined'] = df['combined'].fillna('')

# Check that all values are strings
df['combined'] = df['combined'].astype(str)

# Convert the combined column to lowercase for consistency
df_combined = df.copy()
df_combined['combined'] = df_combined['combined'].str.lower()

# Load embedding model and move to GPU if available
model = SentenceTransformer('all-MiniLM-L6-v2')
if torch.cuda.is_available():
    modle = model.to('cuda')

# Create embeddings for all products once at the start
df['embeddings'] = df['combined'].apply(lambda x: model.encode(x))
df["embedding"] = df.combined.apply(lambda x: model.encode(x))

# 2. Define the Search Function for Gradio
def gradio_search(query):
    # Set the number of results to display
    n = 3

    # Embed the user query
    query_embedding = model.encode(query)

    # Calculate similarity
    df["similarity"] = df.embedding.apply(lambda x: util.cos_sim(x, query_embedding).item())

    # Sort by similarity and return the top 'n' results
    results = df.sort_values("similarity", ascending=False).head(n)
    resultlist = []

    # Collect results in a simple format
    for r in results.index:
        resultlist.append({
            "Product Name": results.Product_Name[r],
            "Score": results.similarity[r],
            "Category": results.Product_Category[r],
            "Price": results["Cleaned Price"][r],
            "Details": results.Product_Details[r],
            "Ingredients": results.Ingredients[r]
        })
    return resultlist

# 3. Maintain History for Multi-Turn Conversations
def chatbot_response(history, query):
    history.append(("User", query))  # Log user input
    results = gradio_search(query)
    
    if len(results) == 0:
        reply = "I couldn't find any matching products. Could you provide more details or rephrase your request?"
    else:
        reply = "\n".join([f"Product: {r['Product Name']}\nCategory: {r['Category']}\nPrice: {r['Price']}\nDetails: {r['Details']}\n" for r in results])

    history.append(("Vinuca", reply))  # Log bot response
    return history, reply

# 4. Set up the Gradio Interface
with gr.Blocks(title="Vinuca AI") as iface:
    gr.Markdown("# Vinuca AI, Your Personal Haircare Assistant")
    chatbot = gr.Chatbot()
    user_input = gr.Textbox(placeholder="Ask me about haircare products...")
    clear = gr.Button("Clear")

    history = []

    def interact(query):
        global history
        history, reply = chatbot_response(history, query)
        return history

    user_input.submit(interact, user_input, chatbot)
    clear.click(lambda: [], None, chatbot)

"""iface = gr.Interface(
    fn=gradio_search,
    inputs="text",
    outputs="json",
    title="Ulta Hair Recommendation Search",
    description="Enter your preferences to find matching products!"
)"""

# 4. Run the App
if __name__ == "__main__":
    iface.launch()

	"""
	import gradio as gr

	def greet(name):
	return "Hello " + name + "!!"

	demo = gr.Interface(fn=greet, inputs="text", outputs="text")
	demo.launch()"""

	import gradio as gr
	import pandas as pd
	import torch
	from sentence_transformers import SentenceTransformer, util
	import google.generativeai as genai
	import pathlib
	import textwrap
	from IPython.display import display
	from IPython.display import Markdown
	import time

	# 1. Initialization - Load Data and Set Up model
	url = 'https://raw.githubusercontent.com/TRASJEPS/Vinicunca_AI_Project1/refs/heads/main/_Data-20241116T221029Z-001/Data/Cleaned%20Data/Combined%20Cleaned%20Data.csv'
	df = pd.read_csv(url)

	# Combine relevant columns
	df["combined"] = (
	"Product Name: " + df.Product_Name.str.strip()+"; Brand: " + df.Brand.str.strip()
	+ "; Category: " + df.Product_Category.str.strip()
	+ "; Details: " + df.Product_Details.str.strip()
	+ "; Ingredients: " + df.Ingredients.str.strip()
	+ "; Price: " + df["Cleaned Price"].str.strip()
	# +"; desc: "+ df.text.str.strip()
	)

	# Ensure the 'combined' column has no NaN or invalid values
	df['combined'] = df['combined'].fillna('')

	# Check that all values are strings
	df['combined'] = df['combined'].astype(str)

	# Convert the combined column to lowercase for consistency
	df_combined = df.copy()
	df_combined['combined'] = df_combined['combined'].str.lower()

	# Load embedding model and move to GPU if available
	model = SentenceTransformer('all-MiniLM-L6-v2')
	if torch.cuda.is_available():
	modle = model.to('cuda')

	# Create embeddings for all products once at the start
	df['embeddings'] = df['combined'].apply(lambda x: model.encode(x))
	df["embedding"] = df.combined.apply(lambda x: model.encode(x))

	# 2. Define the Search Function for Gradio
	def gradio_search(query):
	# Set the number of results to display
	n = 3

	# Embed the user query
	query_embedding = model.encode(query)

	# Calculate similarity
	df["similarity"] = df.embedding.apply(lambda x: util.cos_sim(x, query_embedding).item())

	# Sort by similarity and return the top 'n' results
	results = df.sort_values("similarity", ascending=False).head(n)
	resultlist = []

	# Collect results in a simple format
	for r in results.index:
	resultlist.append({
	"Product Name": results.Product_Name[r],
	"Score": results.similarity[r],
	"Category": results.Product_Category[r],
	"Price": results["Cleaned Price"][r],
	"Details": results.Product_Details[r],
	"Ingredients": results.Ingredients[r]
	})
	return resultlist

	# 3. Maintain History for Multi-Turn Conversations
	def chatbot_response(history, query):
	history.append(("User", query)) # Log user input
	results = gradio_search(query)

	if len(results) == 0:
	reply = "I couldn't find any matching products. Could you provide more details or rephrase your request?"
	else:
	reply = "\n".join([f"Product: {r['Product Name']}\nCategory: {r['Category']}\nPrice: {r['Price']}\nDetails: {r['Details']}\n" for r in results])

	history.append(("Vinuca", reply)) # Log bot response
	return history, reply

	# 4. Set up the Gradio Interface
	with gr.Blocks(title="Vinuca AI") as iface:
	gr.Markdown("# Vinuca AI, Your Personal Haircare Assistant")
	chatbot = gr.Chatbot()
	user_input = gr.Textbox(placeholder="Ask me about haircare products...")
	clear = gr.Button("Clear")

	history = []

	def interact(query):
	global history
	history, reply = chatbot_response(history, query)
	return history

	user_input.submit(interact, user_input, chatbot)
	clear.click(lambda: [], None, chatbot)

	"""iface = gr.Interface(
	fn=gradio_search,
	inputs="text",
	outputs="json",
	title="Ulta Hair Recommendation Search",
	description="Enter your preferences to find matching products!"
	)"""

	# 4. Run the App
	if __name__ == "__main__":
	iface.launch()