Upload 2 files

app.py

@@ -0,0 +1,121 @@
+import os, json, random, logging
+from typing import List
+from dotenv import load_dotenv
+
+from langchain.agents import AgentType, initialize_agent
+from langchain.chat_models import ChatOpenAI
+from langchain.tools import  Tool
+
+from langchain.schema import SystemMessage
+from langchain.agents import OpenAIFunctionsAgent
+from langchain.prompts import MessagesPlaceholder
+from langchain.agents import AgentExecutor, OpenAIFunctionsAgent
+from langchain.chains.conversation.memory import ConversationBufferMemory
+from langchain.chat_models import ChatOpenAI
+from langchain.agents import tool, AgentExecutor, OpenAIFunctionsAgent, AgentType, Agent
+from langchain.schema import SystemMessage
+from langchain.prompts import MessagesPlaceholder
+from langchain.chains.conversation.memory import ConversationBufferMemory
+from langchain.chat_models import ChatOpenAI
+
+from langchain.prompts import ChatPromptTemplate
+from langchain.schema import StrOutputParser
+
+import gradio as gr
+
+load_dotenv()
+
+
+
+llm = ChatOpenAI(temperature=0)
+
+
+
+from typing import List, Dict
+
+@tool
+def describe_popculture_references(references: List) -> Dict:
+    "A tool used to describe pop-culture references as music styles"
+    prompt = ChatPromptTemplate.from_messages([
+        ("system", """You receive a list of pop-culture references (like TV-Shows, films, artists, famous people, etc).
+                      For each reference, write a few words separated by commas which captures the essence of it. Use music styles, sounds and instruments. 
+                      Return a dict with the references as keys and music styles as values.
+                      """),
+        ("human", "{references_list}"),
+    ])
+    runnable = prompt | llm | StrOutputParser()
+    return runnable.invoke({"references_list" : references})
+
+
+@tool
+def extract_popculture_references(input_style: str) -> List:
+    "A tool used to extract pop-culture references from a piece of text"
+    prompt = ChatPromptTemplate.from_messages([
+        ("system", """You detect elements of the pop-culture (like TV-Shows, films, artists, famous people, etc) in the human's input message.
+                      Return a list with these elements only. If there are none, return an empty list.
+                      """),
+        ("human", "{input_style}"),
+    ])
+    runnable = prompt | llm | StrOutputParser()
+    output = runnable.invoke({"input_style" : input_style})
+    return output
+
+@tool
+def call_music_recommendation_api(input : str) -> List[str]:
+    """
+    Calls the music recommendation API
+    """
+    print("Calling music recommendation API: ", input)
+    return {"songs" : [input]}
+
+tools = [describe_popculture_references, extract_popculture_references, call_music_recommendation_api]
+
+
+
+system_message = SystemMessage(content = 
+"""You are an agent which recommends songs based on the style a user gives. 
+You follow the following conversation protocol:
+- You start the conversation by asking the user what style of music they like
+- The user responds with a style of music
+- If there are pop culture references like a movie, a TV show, an artist, a famous person, extract them AND then describe them as music styles.
+- Ask the user if he is ok with the new generated style
+- If the user agrees, call the music recommendation API with this style.
+""")
+
+
+MEMORY_KEY = "chat_history"
+prompt = OpenAIFunctionsAgent.create_prompt(
+    system_message=system_message,
+    extra_prompt_messages=[MessagesPlaceholder(variable_name=MEMORY_KEY)]
+)
+
+memory = ConversationBufferMemory(memory_key=MEMORY_KEY, return_messages=True)
+
+
+agent = OpenAIFunctionsAgent(
+    llm=llm, 
+    tools=tools, 
+    prompt=prompt, 
+    agent=AgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION
+)
+
+agent_executor = AgentExecutor(agent=agent, tools=tools, memory=memory, verbose=True)
+
+
+def inference(message, history):
+#   return agent_executor.run(message)
+    for chunk in agent_executor.stream(message):
+        yield chunk["output"]
+        
+
+gr.ChatInterface(
+    inference,
+    chatbot=gr.Chatbot(height=400),
+    textbox=gr.Textbox(placeholder="Ask me for music recommendations!", container=False, scale=7),
+    description="This AI makes song recommendations based on your music style.",
+    title="Persona Music song recommender",
+    examples=["Recommend me something in Quentin Tarantino reggae style", "Give me songs with calm and relaxing vibes", "I want to listen to something like the movie Inception", "I want music that sounds like Lebron James eating soup"],
+    retry_btn="Retry",
+    clear_btn="Clear",
+    undo_btn = None,
+).queue().launch()

requirements.txt

@@ -0,0 +1,3 @@
+langchain
+openai
+python-dotenv