Replaced Gemma for OpenAI API and commented out visualize_quiz_answers (seems to be breaking the script).

Gemma.py

@@ -3,12 +3,14 @@ from huggingface_hub import login
 import spaces
 import json
 import os
+import openai
 
 __export__ = ["GemmaLLM"]
 
 class GemmaLLM:
     def __init__(self):
         login(token=os.environ.get("GEMMA_TOKEN"))
+        openai.api_key = os.environ.get("OPENAI_API_KEY")
         
         # quant_config = quantization_config.BitsAndBytesConfig(
         #     load_in_8bit=True,
@@ -16,21 +18,23 @@ class GemmaLLM:
         #     llm_int8_has_fp16_weight=False,
         # )
         
-        model_id = "google/gemma-3-4b-it"
+        # model_id = "google/gemma-3-4b-it"
         # model_id = "google/gemma-3n-E4B-it-litert-preview"
         # model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=quant_config)
         # tokenizer = AutoTokenizer.from_pretrained(model_id)
         
-        self.model = pipeline("text-generation", model=model_id)
+        # self.model = pipeline("text-generation", model=model_id)
         
     @spaces.GPU
     def generate(self, message) -> str:
-        outputs = self.model(message, max_new_tokens=1024)[0]["generated_text"]
-        
-        return outputs
+        response = openai.chat.completions.create(
+            model="gpt-4.1",  # You can use "gpt-4" if you have access
+            messages=message)
+        return response.choices[0].message.content
+        # outputs = self.model(message, max_new_tokens=1024)[0]["generated_text"]
+        # return outputs
 
     def _get_summary_message(self, article, num_paragraphs) -> dict:
-        
         summarize = "You are a helpful assistant. Your main task is to summarize articles. You will be given an article that you will generate a summary for. The summary should include all the key points of the article. ONLY RESPOND WITH THE SUMMARY!!!"
 
         summary = f"Summarize the data in the following JSON into {num_paragraphs} paragraph(s) so that it is easy to read and understand:\n"
@@ -43,8 +47,7 @@ class GemmaLLM:
     def get_summary(self, article, num_paragraphs) -> str:
         message = self._get_summary_message(article, num_paragraphs)
         summary = self.generate(message)
-        
-        return summary[2]["content"]
+        return summary
 
     def _get_questions_message(self, summary, num_questions, difficulty) -> dict:
         question = f"""
@@ -62,7 +65,33 @@ class GemmaLLM:
         return message
     
     def get_questions(self, summary, num_questions, difficulty) -> dict:
+        # print("Getting questions message...")
         message = self._get_questions_message(summary, num_questions, difficulty)
+        # print("Generating questions...")
         questions = self.generate(message)
+        # print(questions)
+        questions = questions.strip("```")
+        questions = questions.replace("json\n", "")
+        questions = json.loads(questions)
+        print(questions)
+        return questions
+
+if __name__ == "__main__":
+    gemma = GemmaLLM()
+    summary = gemma.get_summary('''
+    Iran could reach an agreement similar to the 2015 nuclear deal to end its current conflict with Israel, the Iranian foreign minister said Saturday, according to state media.
+
+    “It is clear the Israelis are against diplomacy and do not want the current crisis to be resolved peacefully,” Abbas Araghchi said, according to state-run news agency IRNA.
+
+    “We are, however, certainly prepared to reach a negotiated solution, similar to the one we reached in 2015,” the foreign minister said, according to IRNA.
+
+    Remember: The 2015 deal, formally known as the Joint Comprehensive Plan of Action, required Iran to limit its nuclear program in return for provisions including relief on sanctions. US President Donald Trump withdrew from the landmark agreement in 2018 and Iran has increasingly grown its nuclear program since then.
+
+    More from the foreign minister: “That agreement was the result of two years of tireless negotiations, and when it was signed, it was welcomed by the entire world,” Araghchi said of the 2015 deal, according to IRNA.
 
-        return json.loads(questions[2]["content"].strip("```").replace("json\n", ""))
+    “Diplomacy can be effective — just as it was effective in the past — and can be effective in the future as well. But to return to the path of diplomacy, the aggression must be halted,” he continued, according to the news agency.
+    ''', 1)
+    print(summary)
+    questions = gemma.get_questions(summary, 3, "Easy")
+    print(json.dumps(questions, indent=4))
+    

app.py

@@ -3,7 +3,7 @@ import random
 import gradio as gr
 from News import News
 from Gemma import GemmaLLM
-import matplotlib.pyplot as plt
+# import matplotlib.pyplot as plt
 
 # %%
 class Cooldown:
@@ -123,9 +123,6 @@ with gr.Blocks() as demo:
         
         options = random.shuffle([answer] + false_answers)
         
-        print("Question:", question)
-        print(f"Formatted options: {options}")
-        
         return question, options, answer
     
     def hide_quiz(): ################################### Hide quiz-related components
@@ -142,12 +139,8 @@ with gr.Blocks() as demo:
         quiz = [(question, random.sample(distractors + [answer], 4), answer) for question, distractors, answer in quiz]
         questions, options, answers = zip(*quiz) if quiz else ([], [], [])
         
-        print("options", len(options))
-        
-        quiz = [gr.Radio(label=f"{i + 1}: {questions[i]}", choices=options[i], visible=True) for i in range(len(mcqs))]\
-             + [gr.Radio(visible=False) for _ in range(10 - len(mcqs))]
-        
-        print("quiz", len(quiz))
+        quiz = [gr.Radio(label=f"{i + 1}: {questions[i]}", choices=options[i], visible=True) for i in range(len(mcqs))]
+        quiz += [gr.Radio(visible=False) for _ in range(10 - len(mcqs))]
         
         submit = gr.Button("Submit Answers", interactive=bool(answers), visible=True)
         
@@ -271,25 +264,25 @@ with gr.Blocks() as demo:
     
 demo.launch()
 
-def visualize_quiz_answers(answers, *quiz_items):
-    """
-    Visualization of correct/incorrect answers from the quiz
-    """
-    if not answers:
-        return None
+# def visualize_quiz_answers(answers, *quiz_items):
+#     """
+#     Visualization of correct/incorrect answers from the quiz
+#     """
+#     if not answers:
+#         return None
 
-    correct = 0
+#     correct = 0
 
-    for user_ans, question in zip(answers, quiz_items):
-        if user_ans == question["correct_answer"]:
-            correct += 1
+#     for user_ans, question in zip(answers, quiz_items):
+#         if user_ans == question["correct_answer"]:
+#             correct += 1
 
-    incorrect = len(answers) - correct
+#     incorrect = len(answers) - correct
 
-    fig, ax = plt.subplots()
-    ax.bar(["Correct", "Incorrect"], [correct, incorrect])
-    ax.set_ylabel("Questions")
-    ax.set_title("Quiz Score Summary")
+#     fig, ax = plt.subplots()
+#     ax.bar(["Correct", "Incorrect"], [correct, incorrect])
+#     ax.set_ylabel("Questions")
+#     ax.set_title("Quiz Score Summary")
 
-    return fig
+#     return fig