| import os | |
| from pathlib import Path | |
| import PIL.Image | |
| import google.generativeai as genai | |
| def configure_genai(api_key): | |
| genai.configure(api_key=api_key) | |
| def generate_content(model, prompt, stream=True): | |
| response = model.generate_content(prompt, stream=stream) | |
| response.resolve() | |
| return response.text | |
| def gemini(transcript, question): | |
| print(transcript, question) | |
| configure_genai(os.environ['GOOGLE_API_KEY']) | |
| # Create GenerativeModel instance | |
| model = genai.GenerativeModel('gemini-pro') | |
| # Generate content using the model and image | |
| prompt = [f""" | |
| Transcript: | |
| ``` | |
| {transcript} | |
| ``` | |
| Provided is a video transcript enclosed within triple backticks. Your task is to respond to questions that are either based on or directly related to the content of the video transcript. If the question does not pertain to or is not in the context of the video transcript, please reply with "Please ask questions related to the video only." | |
| Note: | |
| - Do not include `video transcript` in your response, refer it as `video`. | |
| Question: {question} | |
| """] | |
| print("prompt", prompt) | |
| response_text = generate_content(model, prompt) | |
| return response_text | |
| # Optionally display as Markdown | |