Update app.py

app.py

@@ -1,17 +1,25 @@
 #https://python.langchain.com/docs/how_to/configure/
 import gradio as gr
+import os
 from langchain_openai import ChatOpenAI
 from langchain_core.prompts import PromptTemplate
 from langchain_core.runnables import ConfigurableField
 
 def process_with_config(topic, temperature, mode, api_key):
     try:
-        # Initialize configurable model
-        model = ChatOpenAI(temperature=0).configurable_fields(
+        # Set API key
+        os.environ["OPENAI_API_KEY"] = api_key
+        
+        # Initialize configurable model with GPT-4o-mini
+        model = ChatOpenAI(
+            model="gpt-4o-mini",  # Specifically using GPT-4o-mini
+            openai_api_key=api_key,
+            temperature=0
+        ).configurable_fields(
             temperature=ConfigurableField(
                 id="llm_temperature",
                 name="LLM Temperature",
-                description="Temperature for response generation"
+                description="Temperature for GPT-4o-mini response generation"
             )
         )
         
@@ -21,7 +29,8 @@ def process_with_config(topic, temperature, mode, api_key):
         ).configurable_alternatives(
             ConfigurableField(id="prompt"),
             default_key="joke",
-            poem=PromptTemplate.from_template("Write a poem about {topic}")
+            poem=PromptTemplate.from_template("Write a poem about {topic}"),
+            joke=PromptTemplate.from_template("Tell me a joke about {topic}")
         )
         
         # Create chain
@@ -39,20 +48,50 @@ 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"]
 
 # Create Gradio interface
 demo = gr.Interface(
     fn=process_with_config,
     inputs=[
-        gr.Textbox(label="Topic", placeholder="Enter a topic..."),
-        gr.Slider(0, 1, value=0.5, label="Temperature"),
-        gr.Radio(["joke", "poem"], label="Mode", value="joke"),
-        gr.Textbox(label="OpenAI API Key", type="password")
+        gr.Textbox(
+            label="Topic",
+            placeholder="Enter a topic...",
+            lines=1
+        ),
+        gr.Slider(
+            minimum=0,
+            maximum=1,
+            value=0.5,
+            step=0.1,
+            label="Temperature (GPT-4o-mini)"
+        ),
+        gr.Radio(
+            choices=["joke", "poem"],
+            label="Mode",
+            value="joke"
+        ),
+        gr.Textbox(
+            label="OpenAI API Key",
+            placeholder="Enter your OpenAI API key",
+            type="password"
+        )
     ],
-    outputs=gr.Textbox(label="Generated Response"),
-    title="LangChain Configuration Demo",
-    description="Generate content with configurable temperature and mode"
+    outputs=gr.Textbox(
+        label="Generated Response",
+        lines=5
+    ),
+    title="🤖 GPT-4o-mini Configuration Demo",
+    description="Generate content using GPT-4o-mini with configurable temperature and mode"
 )
 
+# Launch the application
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        share=False,
+        server_name="0.0.0.0",
+        server_port=7860
+    )