Update app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
-from langchain.llms import OpenAI as LangChainOpenAI
+# from langchain.llms import OpenAI as LangChainOpenAI
 from langchain.prompts import PromptTemplate
 from langchain_core.runnables import RunnablePassthrough
 from langchain.chains import LLMChain
+from langchain_openai import ChatOpenAI
+
 import requests
 import numpy as np
 import os
@@ -14,7 +16,9 @@ from scipy.io import wavfile
 from pydub import AudioSegment
 
 # Initialize OpenAI clients
-llm = LangChainOpenAI(model_name="gpt-4o-mini")
+
+llm = ChatOpenAI(temperature=0, model_name="gpt-4o-mini")
+# llm = LangChainOpenAI(model_name="gpt-4o-mini")
 openai_client = OpenAI()
 
 # Create a prompt template for the story
@@ -25,8 +29,8 @@ Topic: {topic}
 Story:
 """
 
-story_prompt = PromptTemplate(template=story_template, input_variables=["topic"])
-# story_chain = (RunnablePassthrough() | story_prompt | llm)
+story_prompt = PromptTemplate.from_template(story_template)
+story_chain = (RunnablePassthrough() | story_prompt | llm)
 
 # Create a prompt template for the music based on the story prompt
 music_template = """
@@ -43,11 +47,12 @@ Story Topic: {topic}
 Music Prompt:
 """
 
-music_prompt = PromptTemplate(template=music_template, input_variables=["topic"])
+music_prompt = PromptTemplate.from_template(music_template)
 
 # Create LLMChains
-story_chain = LLMChain(llm=llm, prompt=story_prompt)
-music_chain = LLMChain(llm=llm, prompt=music_prompt)
+# story_chain = LLMChain(llm=llm, prompt=story_prompt)
+# music_chain = LLMChain(llm=llm, prompt=music_prompt)
+music_chain = (RunnablePassthrough() | music_prompt | llm)
 
 # Configuration for music generation
 def get_config():
@@ -151,7 +156,7 @@ def generate_story_and_music(topic):
     with tempfile.TemporaryDirectory() as temp_dir:
         try:
             # Generate the story
-            story = story_chain.run(topic)
+            story = story_chain.invoke({"topic":topic}).content
             
             # Ensure the story has 5 paragraphs
             paragraphs = story.split('\n\n')
@@ -166,7 +171,7 @@ def generate_story_and_music(topic):
 
         try:
             # Generate the music prompt based on the topic
-            final_music_prompt = music_chain.run(topic).strip()
+            final_music_prompt = music_chain.invoke({"topic":topic}).content.strip()
         except Exception as e:
             print(f"Error generating music prompt: {e}")
             final_music_prompt = "Piano and strings creating a melancholic yet hopeful mood with moderate energy in a contemporary classical style"