Merge pull request #34 from GYFX35/feat/language-learning-role-11671587927192561329

Add Language Learning Role to AI Agent

Author: GYFX35

app.py

@@ -834,6 +834,17 @@ def messenger_code_endpoint():
     return jsonify({"status": "success", "message": message})
 
 
+@app.route('/api/v1/learn/language', methods=['POST'])
+@require_api_key
+def learn_language_endpoint():
+    data = request.get_json()
+    prompt = data.get('prompt')
+    if not prompt:
+        return jsonify({"error": _("Prompt is required")}), 400
+    message = google_ai.learn_language(prompt)
+    return jsonify({"status": "success", "message": message})
+
+
 @app.route('/api/register', methods=['POST'])
 def register():
     data = request.get_json()

google_ai.py

@@ -318,3 +318,29 @@ def generate_messenger_code(prompt: str) -> str:
     except Exception as e:
         print(f"Error generating Messenger code with Vertex AI: {e}")
         return f"Error: {e}"
+
+
+def learn_language(prompt: str) -> str:
+    """
+    Provides language learning assistance using Vertex AI.
+    """
+    model = GenerativeModel("gemini-1.5-flash")
+
+    generation_prompt = f"""
+    You are a friendly and patient language tutor. Your task is to help a user learn a new language based on their request.
+
+    User Request:
+    ---
+    {prompt}
+    ---
+
+    Provide a clear, concise, and helpful response. You can provide translations, explanations of grammar, cultural context, or practice exercises.
+    """
+
+    try:
+        response = model.generate_content(generation_prompt)
+        return response.text.strip()
+
+    except Exception as e:
+        print(f"Error providing language learning assistance with Vertex AI: {e}")
+        return f"Error: {e}"