Update app.py

app.py

@@ -1,33 +1,37 @@
 from transformers import pipeline
 import gradio as gr
 
-# Load a better AI model
-generator = pipeline("text2text-generation", model="google/flan-t5-base")
+# Load a small model for faster response
+generator = pipeline("text2text-generation", model="google/flan-t5-small")
 
-def prompting_coach(prompt):
-    response = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
+def turtle_prompt_coach(prompt):
+    # Generate turtle code
+    response = generator(f"Write turtle graphics code to: {prompt}", max_length=200, num_return_sequences=1)[0]['generated_text']
 
+    # Feedback rules
     tips = []
-    if len(prompt.split()) < 5:
-        tips.append("🟡 Try making your prompt longer for better results.")
-    if not prompt.endswith("?") and prompt.lower().startswith(("what", "how", "why")):
-        tips.append("🟡 Try ending with a question mark (?) if you're asking something.")
-    if any(word in prompt.lower() for word in ["cat", "dog", "story"]):
-        tips.append("✅ Nice! Specific topics help the AI stay focused.")
+    if "star" not in prompt.lower():
+        tips.append("🟡 Tip: Mention that you want a **star** shape.")
+    if "5" not in prompt and "five" not in prompt.lower():
+        tips.append("🟡 Tip: How many points? Try saying '5-pointed star'.")
+    if "angle" not in prompt.lower() and "degree" not in prompt.lower():
+        tips.append("🟡 Tip: Include angles or directions, like 'turn right 144 degrees'.")
+    if "turtle" not in prompt.lower():
+        tips.append("🟡 Tip: Mention 'turtle graphics' so the AI knows the style of code you want.")
 
-    feedback = "\n".join(tips) if tips else "✅ Great prompt! Let's see what the AI thinks."
+    feedback = "\n".join(tips) if tips else "✅ Nice prompt! You gave clear instructions for turtle to draw a star."
 
-    return response, feedback
+    return response.strip(), feedback
 
 gr.Interface(
-    fn=prompting_coach,
+    fn=turtle_prompt_coach,
     inputs="text",
     outputs=["text", "text"],
-    title="🎓 Prompting Coach",
-    description="Type a prompt and get an AI-generated response + tips to improve your prompt!",
+    title="🌟 Turtle Graphics Prompting Coach",
+    description="Describe the star you want turtle to draw. Get code + tips to improve your prompt!",
     examples=[
-        ["Tell me a story about a robot in space."],
-        ["What is AI?"],
-        ["Write a poem about pizza."]
+        ["Draw a 5-pointed star using turtle graphics"],
+        ["Use turtle to draw a star with angles"],
+        ["Make a star using forward and right commands"],
     ]
 ).launch()