update: prompt and response refinement

app.py

@@ -1,12 +1,10 @@
 import streamlit as st
 
 from src.pipeline.main import LearnableAI
-import streamlit as st
-from src.pipeline.main import LearnableAI
 
 
 def main():
-    st.title("LearnableAI Interface")
+    st.title("LearnableAI Interface(Use Case 1)")
 
     learnable_ai = LearnableAI()
 
@@ -14,14 +12,15 @@ def main():
     interest = st.text_input("Enter your interest:", "")
     difficulty = st.selectbox(
         "Select difficulty level:",
-        ("very simple", "simple", "hard", "very hard")
+        ("Beginner", "Easy", "Challenging", "Expert",
+         "Starter", "Basic", "Tough", "Master")
     )
 
     if st.button("Process with LearnableAI"):
         if words and interest:  # Only show warning if fields are empty
             try:
                 result = learnable_ai.word_to_sentence(
-                    words=list(words),
+                    words=list(words.split(",")),
                     interest=interest,
                     difficulties=difficulty
                 )
@@ -36,4 +35,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

src/services/use_case_one/word_to_sentence.py

@@ -3,9 +3,11 @@ project @ LearnableAI
 created @ 2025-01-16
 author  @ github.com/ishworrsubedii
 """
-import json
+import os
 from typing import List, Dict
 
+from groq import Groq
+
 
 class UseCaseOne:
     def __init__(self, client):
@@ -13,54 +15,74 @@ class UseCaseOne:
 
     def generate_educational_content(self, interest: str, difficulties: str, words: List[str]) -> Dict:
         prompt = f"""
-        As an educational content specialist for children with ADHD, create engaging learning material that adapts to different difficulty levels.
-
-        Interest area: {interest}
-        Difficulty level: {difficulties} (simple/hard)
-        Words to use: {', '.join(words)}
-
-        Please provide the following in a JSON format:
-        {{
-            "sentences": {{
-                "{difficulties}": [
-                    "Three very simple sentences using the provided words, focused on basic understanding",
-                    "Each sentence should be short and use straightforward language",
-                    "Connect each sentence to the interest area"
-                ],
-            }},
-            "learning_objectives": [
-                "Three clear objectives that match the chosen difficulty level",
-                "Focus on word usage and comprehension",
-                "Include measurable outcomes"
-            ],
-            "activities": [
-                "Two interactive activities that maintain attention",
-                "Activities should match the difficulty level",
-                "Include clear instructions"
-            ]
-        }}
-
-        Ensure all content is:
-        - Clear and direct
-        - Highly visual and descriptive
-        - Strongly connected to the interest area
-        - Appropriate for the chosen difficulty level
-        - ADHD-friendly (engaging, focused, and interactive)
-        - Uses concrete examples and real-world connections
-
-        For simple difficulty:
-        - Use basic sentence structures
-        - Limit to 5-7 words per sentence
-        - Focus on direct cause and effect
-        - Use familiar contexts
-
-        For hard difficulty:
-        - Use compound sentences
-        - Include 8-12 words per sentence
-        - Add descriptive details
-        - Introduce new contexts within the interest area
-
-        Return ONLY a valid JSON object with the structure shown above. Do not include any additional text or explanations.
+        # 🎯 Learning Content Generator
+
+        ## 📝 Input Parameters
+        - **Interest Area:** {interest} 
+        - **Target Words:** {', '.join(words)}
+        - **Difficulty Level:** {difficulties}
+
+        ## ✨ Content Requirements
+
+        ### 1. 📖 Creative Sentences
+        - Must use target words naturally
+        - Must relate to {interest}
+        - Must match {difficulties} difficulty level
+        - Should be engaging and educational
+
+        ### 2. 🤔 Quiz Questions
+        - Must test word understanding
+        - Must be multiple choice
+        - Must have clear correct answers
+        - Must relate to {interest}
+
+        ### 3. 🎨 Image Generation Prompt
+        - Must incorporate target words
+        - Must be detailed for AI generation
+        - Must be creative and engaging
+        - Must relate to {interest}
+
+        ### 4. 🎮 Activities
+        - Must be simple and engaging
+        - Must reinforce target words
+        - Must match {difficulties} level
+        - Must relate to {interest}
+
+        ## 🎁 Output Format
+
+        ### 📚 Sentences
+        1. [First creative sentence]
+        2. [Second creative sentence]
+        3. [Third creative sentence]
+
+        ### ❓ Quiz
+        **Question 1**
+        [Question text]
+        - A) [Option]
+        - B) [Option]
+        - C) [Option]
+        - D) [Option]
+
+        *Correct Answer: [A/B/C/D]*
+
+        **Question 2**
+        [Question text]
+        - A) [Option]
+        - B) [Option]
+        - C) [Option]
+        - D) [Option]
+
+        *Correct Answer: [A/B/C/D]*
+
+        ### 🎯 Activities
+        1. [Detailed description of first activity]
+        2. [Detailed description of second activity]
+
+        ### 🖼️ Image Generation Prompt
+        [Detailed scene description incorporating target words]
+
+        ---
+        *Note: All content should be accessible and engaging for children with learning difficulties.* ✨
         """
 
         try:
@@ -75,14 +97,12 @@ class UseCaseOne:
                     }],
                 model="mixtral-8x7b-32768",
                 temperature=1.0,  # if increase temperature, it will generate more creative content
-                max_tokens=1000,
+                max_tokens=2000,
                 top_p=1,
                 stream=False
             )
-            print(completion.choices[0].message.content)
 
-            content = json.loads(completion.choices[0].message.content)
-            return content
+            return completion.choices[0].message.content
 
         except Exception as e:
             return {
@@ -90,18 +110,14 @@ class UseCaseOne:
                 "status": "failed"
             }
 
-    def format_response(self, content: Dict) -> Dict:
-        return {
-            "educational_content": {
-                "sentences": content.get("sentences", []),
-                "image_prompts": content.get("image_prompts", []),
-                "learning_objectives": content.get("learning_objectives", []),
-                "difficulty_level": content.get("difficulty_level", ""),
-                "activities": content.get("attention_maintaining_activities", [])
-            },
-            "metadata": {
-                "status": "success",
-                "model": "mixtral-8x7b-32768",
-                "version": "1.0"
-            }
-        }
+
+if __name__ == '__main__':
+    client = Groq(api_key=os.getenv
+    ('GROQ_API_KEY'))
+    use_case_one = UseCaseOne(client=client)
+    result = use_case_one.generate_educational_content(
+        interest="spiderman",
+        difficulties="simple",
+        words=["apple", "banana", "orange"]
+    )
+    print(result)