Update app.py

app.py

@@ -12,7 +12,7 @@ def process_with_config(topic, temperature, mode, api_key):
         
         # Initialize configurable model with GPT-4o-mini
         model = ChatOpenAI(
-            model="gpt-4o-mini",  # Specifically using GPT-4o-mini
+            model="gpt-4o-mini",
             openai_api_key=api_key,
             temperature=0
         ).configurable_fields(
@@ -23,24 +23,27 @@ def process_with_config(topic, temperature, mode, api_key):
             )
         )
         
-        # Create configurable prompt
+        # Create configurable prompt with unique keys
         prompt = PromptTemplate.from_template(
-            "Tell me a {mode} about {topic}"
+            "Generate a {mode} about {topic}"
         ).configurable_alternatives(
-            ConfigurableField(id="prompt"),
-            default_key="joke",
-            poem=PromptTemplate.from_template("Write a poem about {topic}"),
-            joke=PromptTemplate.from_template("Tell me a joke about {topic}")
+            ConfigurableField(id="content_type"),
+            default_key="default",
+            alternatives={
+                "default": PromptTemplate.from_template("Tell me a joke about {topic}"),
+                "poetry": PromptTemplate.from_template("Write a poem about {topic}")
+            }
         )
         
         # Create chain
         chain = prompt | model
         
-        # Configure and run
+        # Configure and run with correct mode mapping
+        mode_mapping = {"joke": "default", "poem": "poetry"}
         response = chain.with_config(
             configurable={
                 "llm_temperature": float(temperature),
-                "prompt": mode
+                "content_type": mode_mapping[mode]
             }
         ).invoke({"topic": topic})
         
@@ -49,7 +52,6 @@ def process_with_config(topic, temperature, mode, api_key):
     except Exception as e:
         return f"Error: {str(e)}"
     finally:
-        # Clear API key from environment for security
         if "OPENAI_API_KEY" in os.environ:
             del os.environ["OPENAI_API_KEY"]
 
@@ -88,10 +90,5 @@ demo = gr.Interface(
     description="Generate content using GPT-4o-mini with configurable temperature and mode"
 )
 
-# Launch the application
 if __name__ == "__main__":
-    demo.launch(
-        share=False,
-        server_name="0.0.0.0",
-        server_port=7860
-    )
+    demo.launch()