Update utils.py

utils.py

@@ -1,40 +1,54 @@
 import openai
-import os
+from pathlib import Path
+from PIL import Image
+from io import BytesIO
+import base64
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Make sure your OPENAI_API_KEY is set in environment variables
 
 def generate_lesson(topic):
-    """Generates a short, beginner-friendly code lesson for the chosen topic."""
-    prompt = f"Write a clear, beginner-friendly coding lesson on {topic}. Include a short example with code comments."
-    
+    """
+    Generates lesson text for a given topic.
+    Returns a tuple: (file_path, lesson_text)
+    """
+    tutorial_prompt = f"""
+    Create a beginner-friendly coding tutorial for '{topic}'.
+    Include explanations and a minimal Python/JS/HTML/CSS code snippet relevant to the topic.
+    End with a short practice section.
+    """
     response = openai.chat.completions.create(
         model="gpt-4o-mini",
-        messages=[
-            {"role": "system", "content": "You are an expert code instructor who writes short, readable lessons."},
-            {"role": "user", "content": prompt}
-        ]
+        messages=[{"role": "user", "content": tutorial_prompt}],
+        temperature=0.7,
     )
+    lesson_text = response.choices[0].message.content.strip()
 
-    text = response.choices[0].message.content.strip()
-    file_path = f"/tmp/{topic.replace(' ', '_')}_lesson.txt"
-    
-    with open(file_path, "w", encoding="utf-8") as f:
-        f.write(text)
-
-    return file_path, text
+    # Save lesson to a .txt file
+    output_path = Path("outputs") / f"{topic.replace(' ', '_')}.txt"
+    output_path.parent.mkdir(exist_ok=True)
+    with open(output_path, "w", encoding="utf-8") as f:
+        f.write(lesson_text)
+    return str(output_path), lesson_text
 
 
 def generate_image(topic):
-    """Generates a minimal dark terminal-style code screenshot for the lesson topic."""
-    image_prompt = (
-        f"A minimal terminal screenshot showing coding related to {topic}. "
-        "Dark background, bright syntax highlighting, subtle glow, realistic developer aesthetic."
+    """
+    Generates a terminal-style image snippet for a lesson topic.
+    """
+    prompt = (
+        f"A minimal terminal screenshot showing coding for a lesson on '{topic}'. "
+        "Show a code snippet or highlighted portion of code, dark background, light text, Consolas font."
     )
-
     response = openai.images.generate(
         model="gpt-image-1",
-        prompt=image_prompt,
+        prompt=prompt,
         size="1024x1024"
     )
 
-    return response.data[0].url
+    image_data = response.data[0].b64_json
+    image = Image.open(BytesIO(base64.b64decode(image_data)))
+
+    output_path = Path("outputs") / f"{topic.replace(' ', '_')}.png"
+    output_path.parent.mkdir(exist_ok=True)
+    image.save(output_path)
+    return str(output_path)