Update app.py

app.py

@@ -5,10 +5,17 @@ from transformers import pipeline
 generator = pipeline('text-generation', model='distilgpt2')
 
 def generate_character_description(name, gender, trait, superpower):
-    prompt = f"Generate a detailed description of a character named {name}. They are {gender}, {trait}, and their superpower is {superpower}. Provide a short story or background for this character."
-    # Generate the description based on the prompt
-    description = generator(prompt, max_length=100)[0]['generated_text']
-    return description
+    # Construct the prompt with specific instructions for detailed descriptions
+    prompt = (f"Create a game senario of this character description. Name: {name}. "
+              f"Gender: {gender}. Personality Trait: {trait}. "
+              f"Superpower: {superpower}. Provide a short story or background for this character.")
+    
+    # Generate the description
+    results = generator(prompt, max_length=150, num_return_sequences=1)
+    description = results[0]['generated_text']
+    
+    # Clean up the output by removing the prompt part from the result
+    return description[len(prompt):].strip()
 
 # Gradio interface
 iface = gr.Interface(