Update app.py

app.py

@@ -3,85 +3,66 @@ import gradio as gr
 import os
 from langchain_openai import ChatOpenAI
 from langchain_core.prompts import ChatPromptTemplate
-from langchain_core.runnables import ConfigurableField
 
-def process_with_config(topic, temperature, mode, api_key):
+def generate_response(prompt, temperature, api_key):
     try:
-        # Set API key
-        os.environ["OPENAI_API_KEY"] = api_key
-        
-        # Initialize model with configurable temperature
-        model = ChatOpenAI(
-            model="gpt-4o-mini",
+        # Initialize model with given temperature
+        llm = ChatOpenAI(
+            model="gpt-3.5-turbo",  # Using 3.5-turbo as it's more reliable
+            temperature=float(temperature),
             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 = prompt | model
+        # Create simple prompt
+        template = ChatPromptTemplate.from_messages([
+            ("system", "You are a creative assistant."),
+            ("user", "Generate a creative description of {prompt}")
+        ])
         
-        # Execute with configuration
-        response = chain.with_config(
-            configurable={
-                "temperature": float(temperature)
-            }
-        ).invoke({"topic": topic})
+        # Create and run chain
+        chain = template | llm
+        response = chain.invoke({"prompt": prompt})
         
         return response.content
         
     except Exception as e:
         return f"Error: {str(e)}"
-    finally:
-        if "OPENAI_API_KEY" in os.environ:
-            del os.environ["OPENAI_API_KEY"]
 
 # Create Gradio interface
 demo = gr.Interface(
-    fn=process_with_config,
+    fn=generate_response,
     inputs=[
         gr.Textbox(
-            label="Topic",
-            placeholder="Enter a topic...",
-            lines=1
+            label="Prompt",
+            placeholder="Example: 'a blue elephant'",
+            value="a blue elephant"
         ),
         gr.Slider(
             minimum=0,
-            maximum=1,
-            value=0.5,
+            maximum=2,
+            value=0.7,
             step=0.1,
-            label="Temperature (GPT-4o-mini)"
-        ),
-        gr.Radio(
-            choices=["joke", "poem"],
-            label="Style",
-            value="joke"
+            label="Temperature (0: Focused, 2: More Creative)"
         ),
         gr.Textbox(
             label="OpenAI API Key",
-            placeholder="Enter your OpenAI API key",
             type="password"
         )
     ],
-    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 style"
+    outputs=gr.Textbox(label="Generated Response", lines=5),
+    title="🎨 Temperature Effect Demo",
+    description="""
+    Try the same prompt with different temperatures:
+    - Temperature 0: More focused, consistent responses
+    - Temperature 0.7: Balanced creativity
+    - Temperature 2: More random, creative responses
+    
+    Example prompt: 'a blue elephant'
+    - T=0: "A blue elephant is a fictional creature with blue-colored skin..."
+    - T=0.7: "Imagine a majestic pachyderm with sapphire-tinted hide..."
+    - T=2: "Dancing through cotton candy clouds, this azure giant..."
+    """
 )
 
 if __name__ == "__main__":
-    demo.launch(share=False)
+    demo.launch()