ai_model/callGeminiModel.py

import google.generativeai as genai
from langchain.prompts import PromptTemplate

# Configure Gemini
api_key = "AIzaSyCWj2sHInMricyC2frSg3uwUgsd_QOGkLA"
genai.configure(api_key=api_key)
model = genai.GenerativeModel(model_name="gemini-1.5-flash-latest")

# Define the Prompt Template
log_prompt = PromptTemplate(
    input_variables=["logs", "query"],
    template=(
        """You are an expert log analyzer. Analyze the system logs provided below. 
        Return only precise and concise answers to the questions asked, formatted clearly.

        Logs: {logs}
        User's Query: {query}

        Be concise and direct in your responses."""
    ),
)


def generate_ai_response(user_query, logs):
    formatted_prompt = log_prompt.format(logs=logs, query=user_query)
    response = model.generate_content([formatted_prompt])
    return response.text