Update app.py

app.py

@@ -1,8 +1,12 @@
+import os
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import torch
 import spaces
 
+# Create the necessary directories
+os.makedirs('.gradio/cached_examples/17', exist_ok=True)
+
 def get_model_name(language):
     """Map language choice to the corresponding model."""
     model_mapping = {
@@ -26,50 +30,52 @@ def load_model(model_name):
         tokenizer=tokenizer,
         return_full_text=False,
         max_new_tokens=500,
-        do_sample=False
+        do_sample=True,  # Enable sampling for more creative outputs
+        top_k=50,        # Control diversity
+        top_p=0.95       # Control diversity
     )
     return generator
+
 @spaces.GPU
-def generate_plant_info(plant_name, language):
+def generate_kids_story(character, setting, language):
     model_name = get_model_name(language)
     generator = load_model(model_name)
 
     # Define prompt for the AI model
     if language == "English":
-        prompt = (f"Provide detailed information about {plant_name}. "
-                  f"Include its scientific name, growing conditions (light, water, soil type), "
-                  f"common uses, and care tips.")
+        prompt = (f"Write a short story for kids about a character named {character} who goes on an adventure in {setting}. "
+                  "Make it fun, engaging, and suitable for children.")
     else:
-        prompt = (f"قدم معلومات مفصلة عن {plant_name}. "
-                  f"اذكر اسمه العلمي، وظروف نموه (الضوء، الماء، نوع التربة)، "
-                  f"استخداماته الشائعة، ونصائح العناية به.")
+        prompt = (f"اكتب قصة قصيرة للأطفال عن شخصية اسمها {character} التي تذهب في مغامرة في {setting}. "
+                  "اجعلها ممتعة وجذابة ومناسبة للأطفال.")
 
     messages = [{"role": "user", "content": prompt}]
     output = generator(messages)
     return output[0]["generated_text"]
 
-
 # Create Gradio interface
 demo = gr.Interface(
-    fn=generate_plant_info,
+    fn=generate_kids_story,
     inputs=[
-        gr.Textbox(placeholder="Enter plant name (e.g., Lavender, Aloe Vera)...", label="Plant Name"),
+        gr.Textbox(placeholder="Enter a character name (e.g., Benny the Bunny)...", label="Character Name"),
+        gr.Textbox(placeholder="Enter a setting (e.g., a magical forest)...", label="Setting"),
         gr.Dropdown(
-            choices=["English", "Arabic"],  # Users choose the language, not the model name
+            choices=["English", "Arabic"],
             label="Choose Language",
             value="English"  # Default to English
         )
     ],
-    outputs=gr.Textbox(label="Plant Information"),
-    title="🌿 AI Plant Guide - English & Arabic 🌿",
-    description="Enter a plant name, and AI will provide detailed information about it in English or Arabic.",
+    outputs=gr.Textbox(label="Kids' Story"),
+    title="📖 AI Kids' Story Generator - English & Arabic 📖",
+    description="Enter a character name and a setting, and AI will generate a fun short story for kids in English or Arabic.",
     examples=[
-        ["Lavender", "English"],
-        ["اللافندر", "Arabic"],
-        ["Tulip", "English"],
-        ["الصبار", "Arabic"]
+        ["Benny the Bunny", "a magical forest", "English"],
+        ["علي البطل", "غابة سحرية", "Arabic"],
+        ["Lila the Ladybug", "a garden full of flowers", "English"],
+        ["ليلى الجنية", "حديقة مليئة بالأزهار", "Arabic"]
     ],
     theme="default",
 )
 
+# Launch the Gradio app
 demo.launch()