Update app.py

app.py

@@ -2,7 +2,7 @@
 import gradio as gr
 import os
 from langchain_openai import ChatOpenAI
-from langchain_core.prompts import PromptTemplate
+from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.runnables import ConfigurableField
 
 def process_with_config(topic, temperature, mode, api_key):
@@ -10,40 +10,43 @@ def process_with_config(topic, temperature, mode, api_key):
         # Set API key
         os.environ["OPENAI_API_KEY"] = api_key
         
-        # Initialize configurable model with GPT-4o-mini
+        # Initialize model
         model = ChatOpenAI(
             model="gpt-4o-mini",
             openai_api_key=api_key,
             temperature=0
-        ).configurable_fields(
-            temperature=ConfigurableField(
-                id="llm_temperature",
-                name="LLM Temperature",
-                description="Temperature for GPT-4o-mini response generation"
-            )
         )
         
-        # Create configurable prompt with unique keys
-        prompt = PromptTemplate.from_template(
-            "Generate a {mode} about {topic}"
-        ).configurable_alternatives(
-            ConfigurableField(id="content_type"),
-            default_key="default",
+        # Create base prompt template
+        base_prompt = ChatPromptTemplate.from_messages([
+            ("system", "You are a helpful assistant."),
+            ("user", "Tell me a joke about {topic}")
+        ])
+        
+        # Create poem prompt template
+        poem_prompt = ChatPromptTemplate.from_messages([
+            ("system", "You are a helpful assistant."),
+            ("user", "Write a poem about {topic}")
+        ])
+        
+        # Create configurable prompt
+        configurable_prompt = base_prompt.configurable_alternatives(
+            ConfigurableField(id="style"),
+            default_key="joke",
             alternatives={
-                "default": PromptTemplate.from_template("Tell me a joke about {topic}"),
-                "poetry": PromptTemplate.from_template("Write a poem about {topic}")
+                "joke": base_prompt,
+                "poem": poem_prompt
             }
         )
         
         # Create chain
-        chain = prompt | model
+        chain = configurable_prompt | model
         
-        # Configure and run with correct mode mapping
-        mode_mapping = {"joke": "default", "poem": "poetry"}
+        # Execute with configuration
         response = chain.with_config(
             configurable={
-                "llm_temperature": float(temperature),
-                "content_type": mode_mapping[mode]
+                "style": mode,
+                "temperature": float(temperature)
             }
         ).invoke({"topic": topic})
         
@@ -73,7 +76,7 @@ demo = gr.Interface(
         ),
         gr.Radio(
             choices=["joke", "poem"],
-            label="Mode",
+            label="Style",
             value="joke"
         ),
         gr.Textbox(
@@ -87,7 +90,7 @@ demo = gr.Interface(
         lines=5
     ),
     title="🤖 GPT-4o-mini Configuration Demo",
-    description="Generate content using GPT-4o-mini with configurable temperature and mode"
+    description="Generate content using GPT-4o-mini with configurable temperature and style"
 )
 
 if __name__ == "__main__":