Create qa_engine.py

qa_engine.py

@@ -0,0 +1,25 @@
+import google.generativeai as genai
+import os
+
+# Configure the Gemini API
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+
+# Initialize the Gemini Pro model for text-based tasks
+model = genai.GenerativeModel('gemini-pro')
+
+def get_answer(question, video_data):
+    prompt = f"""
+    Based on the following video information, please answer the question.
+    
+    Video Summary: {video_data['summary']}
+    Extracted Text: {video_data['extracted_text']}
+    Transcription: {video_data['transcription']}
+    Extracted Code: {video_data['extracted_code']}
+    
+    Question: {question}
+    
+    Answer:
+    """
+    
+    response = model.generate_content(prompt)
+    return response.text