extract metadata

backend/config.py

@@ -1,3 +1,26 @@
+language_metadata_extraction_prompt = """
+You are a language learning assistant. Your task is to analyze the user's input and infer their:
+- Native language (use the language of the input as a fallback if unsure)
+- Target language (the one they want to learn)
+- Proficiency level (beginner, intermediate, or advanced)
+
+Respond ONLY with a valid JSON object using the following format:
+
+{
+  "native_language": "<user's native language>",
+  "target_language": "<language the user wants to learn>",
+  "proficiency_level": "<beginner | intermediate | advanced>"
+}
+
+Guidelines:
+- If the user's native language is not explicitly stated, assume it's the same as the language used in the query.
+- If the target language is mentioned indirectly (e.g. "my Dutch isn't great"), infer that as the target language.
+- Make a reasonable guess at proficiency based on clues like "isn't great" → beginner or "I want to improve" → intermediate.
+- If you cannot infer something at all, write "unknown".
+
+Do not include any explanations, comments, or formatting — only valid JSON.
+"""
+
 flashcard_mode_instructions = """
 You are a versatile vocabulary tutor capable of teaching **any language**. Your goal is to help users learn **as fast as possible** by creating highly relevant vocabulary flashcards **directly tied to their stated hobbies, work, or studies.** The learning experience should feel personalized and engaging because it uses their specific context.
 
@@ -121,4 +144,4 @@ Example JSON Output (Illustrative - content will vary based on domain):
 }
 
 Output ONLY the valid JSON object. Do not include introductory text, notes, or any extra formatting.
-"""
+"""

backend/main.py

@@ -21,10 +21,31 @@ class GenerationRequest(BaseModel):
     user_id: int
     query: str
 
+class MetadataRequest(BaseModel):
+    query: str
+
 @app.get("/")
 async def root():
     return {"message": "Welcome to the AI Learning Assistant API!"}
 
+@app.post("/extract/metadata")
+async def extract_metadata(data: MetadataRequest):
+    try:
+        response = await generate_completions.get_completions(
+            data.query,
+            config.language_metadata_extraction_prompt
+        )
+        return JSONResponse(
+            content={
+                "data": response,
+                "type": "language_metadata",
+                "status": "success"
+            },
+            status_code=200
+        )
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
 @app.post("/generate/flashcards")
 async def generate_flashcards(data: GenerationRequest):
     try: