Merge pull request #31 from GYFX35/feat/add-new-ai-roles-7826620215432288341

feat: Add business strategist, IT support, and data scientist roles

Author: GYFX35

app.py

@@ -774,6 +774,39 @@ def automate_script_endpoint():
     return jsonify({"status": "success", "message": message})
 
 
+@app.route('/api/v1/business/strategy', methods=['POST'])
+@require_api_key
+def business_strategy_endpoint():
+    data = request.get_json()
+    prompt = data.get('prompt')
+    if not prompt:
+        return jsonify({"error": _("Prompt is required")}), 400
+    message = google_ai.generate_business_strategy(prompt)
+    return jsonify({"status": "success", "message": message})
+
+
+@app.route('/api/v1/support/it', methods=['POST'])
+@require_api_key
+def it_support_endpoint():
+    data = request.get_json()
+    prompt = data.get('prompt')
+    if not prompt:
+        return jsonify({"error": _("Prompt is required")}), 400
+    message = google_ai.provide_it_support(prompt)
+    return jsonify({"status": "success", "message": message})
+
+
+@app.route('/api/v1/data/analyze', methods=['POST'])
+@require_api_key
+def analyze_data_endpoint():
+    data = request.get_json()
+    prompt = data.get('prompt')
+    if not prompt:
+        return jsonify({"error": _("Prompt is required")}), 400
+    message = google_ai.analyze_data(prompt)
+    return jsonify({"status": "success", "message": message})
+
+
 @app.route('/api/register', methods=['POST'])
 def register():
     data = request.get_json()

google_ai.py

@@ -139,3 +139,78 @@ def generate_promotion_from_content(url: str, content: str) -> str:
     except Exception as e:
         print(f"Error generating promotion from content with Vertex AI: {e}")
         return f"Error: {e}"
+
+def generate_business_strategy(prompt: str) -> str:
+    """
+    Generates a business strategy from a prompt using Vertex AI.
+    """
+    model = GenerativeModel("gemini-1.5-flash")
+
+    generation_prompt = f"""
+    You are an expert business strategist. Your task is to develop a business strategy based on the following user prompt.
+
+    User Prompt:
+    ---
+    {prompt}
+    ---
+
+    The strategy should be comprehensive and include actionable steps.
+    """
+
+    try:
+        response = model.generate_content(generation_prompt)
+        return response.text.strip()
+
+    except Exception as e:
+        print(f"Error generating business strategy with Vertex AI: {e}")
+        return f"Error: {e}"
+
+def provide_it_support(prompt: str) -> str:
+    """
+    Provides IT support for a given issue using Vertex AI.
+    """
+    model = GenerativeModel("gemini-1.5-flash")
+
+    generation_prompt = f"""
+    You are a knowledgeable IT support specialist. Your task is to provide a solution to the following technical issue.
+
+    User Issue:
+    ---
+    {prompt}
+    ---
+
+    Provide a clear, step-by-step solution.
+    """
+
+    try:
+        response = model.generate_content(generation_prompt)
+        return response.text.strip()
+
+    except Exception as e:
+        print(f"Error providing IT support with Vertex AI: {e}")
+        return f"Error: {e}"
+
+def analyze_data(prompt: str) -> str:
+    """
+    Analyzes data and provides insights using Vertex AI.
+    """
+    model = GenerativeModel("gemini-1.5-flash")
+
+    generation_prompt = f"""
+    You are a skilled data scientist. Your task is to analyze the following data and provide insights.
+
+    Data and Request:
+    ---
+    {prompt}
+    ---
+
+    Provide key insights, trends, and conclusions from the data.
+    """
+
+    try:
+        response = model.generate_content(generation_prompt)
+        return response.text.strip()
+
+    except Exception as e:
+        print(f"Error analyzing data with Vertex AI: {e}")
+        return f"Error: {e}"