Update app.py

app.py

@@ -13,9 +13,7 @@ For more information on `huggingface_hub` Inference API support, please check th
 # client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
-async def advancedPromptPipeline(InputPrompt):
-    
-    model="gpt-4o-mini"
+async def advancedPromptPipeline(InputPrompt, model="gpt-4o-mini", temperature=0.0):
     
     if model == "gpt-4o":
         i_cost=5/10**6
@@ -24,7 +22,7 @@ async def advancedPromptPipeline(InputPrompt):
         i_cost=0.15/10**6
         o_cost=0.6/10**6
     
-    enhancer = PromptEnhancer(model)
+    enhancer = PromptEnhancer(model, temperature)
     
     start_time = time.time()
     advanced_prompt = await enhancer.enhance_prompt(InputPrompt, perform_eval=False)
@@ -93,10 +91,11 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 
 demo = gr.Interface(fn=advancedPromptPipeline, 
                     inputs=[
-                        gr.Textbox(label="Input Prompt", lines=2, placeholder="Enter your prompt"),
-                        gr.Slider(minimum=0.0, maximum=1.0, value=0.7, step=0.1, label="Temperature")
+                        gr.Textbox(lines=20, placeholder="Enter your prompt", label="Input Prompt"),
+                        gr.Radio(["gpt-4o-mini", "gpt-4o"], value="gpt-4o-mini", label="Select Model"),
+                        gr.Slider(minimum=0.0, maximum=1.0, value=0.0, step=0.1, label="Temperature")
                         ], 
-                    outputs=[gr.Textbox(label="Advanced Prompt", lines=30)])
+                    outputs=[gr.Textbox(lines=30, label="Advanced Prompt")])
 
 
 if __name__ == "__main__":