Update app.py

app.py

@@ -50,7 +50,7 @@ async def enhance_prompt(request: EnhanceRequest, api_key: str = Depends(verify_
     ]
     
     try:
-        result = pipe(messages, max_length=512, temperature=0.7)
+        result = pipe(messages, max_new_tokens=256, temperature=0.7, do_sample=True)
         enhanced_prompt = result[0]["generated_text"]
         
         if isinstance(enhanced_prompt, list):
@@ -62,10 +62,16 @@ async def enhance_prompt(request: EnhanceRequest, api_key: str = Depends(verify_
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Enhancement failed: {str(e)}")
 
-def enhance_gradio(prompt_text):
+def enhance_for_gradio(prompt_text, api_key):
     if not prompt_text.strip():
         return "Please enter a prompt to enhance."
     
+    if not api_key.strip():
+        return "Please enter your API key."
+    
+    if api_key != API_KEY:
+        return "Invalid API key."
+    
     messages = [
         {"role": "system", "content": SYSTEM_PROMPT},
         {"role": "user", "content": prompt_text}
@@ -85,7 +91,7 @@ def enhance_gradio(prompt_text):
         return f"Enhancement failed: {str(e)}"
 
 iface = gr.Interface(
-    fn=enhance_gradio,
+    fn=enhance_for_gradio,
     inputs=[
         gr.Textbox(
             lines=5,