added support for content length

app/main.py

@@ -45,6 +45,7 @@ class ContentRequest(BaseModel):
     subject: str
     topic: str
     style: str = "normal"  
+    content_length: str = "medium"
 
 TOPIC_KEYWORDS = {
     # Grade 3 Science
@@ -282,7 +283,14 @@ def generate_response_with_rag(prompt, index, embedding_model, documents, settin
     "normal": ""
     }
 
+    content_length_instructions = {
+        "short": "Keep your response brief and concise. Focus on the most essential points only. Provide only 2-3 subtopics, 1-2 activities, and 3-4 practice questions.",
+        "medium": "",  # Default length, no special instructions
+        "long": "Provide a comprehensive and detailed explanation. Include more examples, detailed explanations for each subtopic (at least 4-5 subtopics), 3-4 activities, and 6-8 practice questions."
+    }
+
     instruction = style_instructions.get(settings.get("style", "normal"), "")
+    length_instruction = content_length_instructions.get(settings.get("content_length", "medium"), "")
 
     # Create system prompt
     system_prompt = f"""
@@ -297,9 +305,9 @@ def generate_response_with_rag(prompt, index, embedding_model, documents, settin
         **Respond to all questions and instructions in Swahili.**
 
         Please follow these instructions:
-        * Fully written-out numbers (e.g., “watoto watatu” instead of “3”)
-        * No abbreviations (e.g., “Shule ya Msingi” not “Sh. ya Msingi”)
-        * Swahili-only words or phonetic “Swahilicized” versions for foreign terms
+        * Fully written-out numbers (e.g., "watoto watatu" instead of "3")
+        * No abbreviations (e.g., "Shule ya Msingi" not "Sh. ya Msingi")
+        * Swahili-only words or phonetic "Swahilicized" versions for foreign terms
         * Minimal punctuation—just commas and full stops
         * Quotations, brackets, symbols removed
         * Natural, spoken tone fit for young learners
@@ -307,6 +315,7 @@ def generate_response_with_rag(prompt, index, embedding_model, documents, settin
         IMPORTANT: Make sure at least two [Picture: Image description] are included in the response.
 
     {instruction}
+    {length_instruction}
 
     Context:
     {retrieved_context}
@@ -514,6 +523,9 @@ async def generate_content(request: ContentRequest):
         if request.style not in ["normal", "simple", "creative"]:
             raise HTTPException(status_code=400, detail="Invalid style. Must be 'normal', 'simple', or 'creative'")
         
+        if request.content_length not in ["short", "medium", "long"]:
+            raise HTTPException(status_code=400, detail="Invalid content length. Must be 'short', 'medium', or 'long'")
+        
         # Get relevant topic files
         topic_files = get_topic_files(request.grade, request.subject, request.topic)
         if not topic_files:
@@ -524,7 +536,8 @@ async def generate_content(request: ContentRequest):
             "style": request.style,
             "topic": request.topic,
             "grade": request.grade,
-            "subject": request.subject
+            "subject": request.subject,
+            "content_length": request.content_length
         }
             
         response = generate_response_with_rag(

spaces_app.py

@@ -36,14 +36,15 @@ asyncio.set_event_loop(loop)
 loop.run_until_complete(startup_event())
 
 # Define the Gradio interface
-def generate(grade, subject, topic, style):
+def generate(grade, subject, topic, style, content_length):
     try:
         # Create a ContentRequest object
         request = ContentRequest(
             grade=int(grade),
             subject=subject,
             topic=topic,
-            style=style
+            style=style,
+            content_length=content_length
         )
         
         # Call the generate_content function
@@ -77,6 +78,11 @@ with gr.Blocks(title="Swahili Content Generation") as demo:
                 label="Style",
                 value="normal"
             )
+            content_length = gr.Dropdown(
+                choices=["short", "medium", "long"], 
+                label="Content Length",
+                value="medium"
+            )
             generate_btn = gr.Button("Generate Content")
         
         with gr.Column():
@@ -84,7 +90,7 @@ with gr.Blocks(title="Swahili Content Generation") as demo:
     
     generate_btn.click(
         fn=generate,
-        inputs=[grade, subject, topic, style],
+        inputs=[grade, subject, topic, style, content_length],
         outputs=output
     )