prompt.py

def create_prompt(query: str) -> str:
    """
    Creates a structured prompt for the Gemini model.

    Args:
        query: The user's query.

    Returns:
        A formatted prompt string.
    """
    # You can customize this prompt to guide the model's responses.
    # For example, you can add a persona, context, or few-shot examples.
    prompt = f"""
    You are a helpful and friendly assistant.

    User query: {query}

    Please provide a concise and accurate response.
    """
    return prompt