Update app.py

app.py

@@ -10,42 +10,32 @@ def process_with_config(topic, temperature, mode, api_key):
         # Set API key
         os.environ["OPENAI_API_KEY"] = api_key
         
-        # Initialize model
+        # Initialize model with configurable temperature
         model = ChatOpenAI(
             model="gpt-4o-mini",
-            openai_api_key=api_key,
-            temperature=0
-        )
-        
-        # 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={
-                "joke": base_prompt,
-                "poem": poem_prompt
-            }
+            openai_api_key=api_key
+        ).configurable_fields(
+            temperature=ConfigurableField(id="temperature")
         )
+
+        # Define different prompts based on mode
+        if mode == "joke":
+            prompt = ChatPromptTemplate.from_messages([
+                ("system", "You are a helpful assistant skilled in telling jokes."),
+                ("user", "Tell me a funny joke about {topic}")
+            ])
+        else:
+            prompt = ChatPromptTemplate.from_messages([
+                ("system", "You are a helpful assistant skilled in writing poetry."),
+                ("user", "Write a beautiful poem about {topic}")
+            ])
         
         # Create chain
-        chain = configurable_prompt | model
+        chain = prompt | model
         
         # Execute with configuration
         response = chain.with_config(
             configurable={
-                "style": mode,
                 "temperature": float(temperature)
             }
         ).invoke({"topic": topic})
@@ -94,4 +84,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=False)