Completed Prototype

Gemma.py

@@ -57,12 +57,9 @@ class GemmaLLM:
 
     def get_summary(self, article, num_paragraphs) -> str:
         message = self._get_summary_message(article, num_paragraphs)
-        try:
-            summary = self.generate(message)
-        except Exception as e:
-            summary = str(e)
+        summary = self.generate(message)
         
-        return summary
+        return summary[2]["content"]
 
     def _get_questions_message(self, summary, num_questions, difficulty) -> dict:
         question = f"""
@@ -83,4 +80,4 @@ class GemmaLLM:
         message = self._get_questions_message(summary, num_questions, difficulty)
         questions = self.generate(message)
 
-        return json.loads(questions.strip("```").replace("json\n", ""))
+        return json.loads(questions[2]["content"].strip("```").replace("json\n", ""))