Completed Prototype

app.py

@@ -8,9 +8,9 @@ from Gemma import GemmaLLM
 class Cooldown:
     ...
 
-cooldown = Cooldown() ################################ News fetching cooldown in seconds
-news = News() ######################################## Initialize the News object
-model = GemmaLLM() ################################### Initialize the Gemma model
+cooldown = Cooldown() ################################## News fetching cooldown in seconds
+news = News() ########################################## Initialize the News object
+model = GemmaLLM() ##################################### Initialize the Gemma model
 
 # %%
 with gr.Blocks() as demo:
@@ -20,7 +20,7 @@ with gr.Blocks() as demo:
     ###### State Variables and Components Initialization
     ######
     
-    init = ( ######################################### Initialize the Gradio interface with components and state variables
+    init = ( ########################################### Initialize the Gradio interface with components and state variables
 
         gr.Markdown("## News Articles"),
     
@@ -52,7 +52,7 @@ with gr.Blocks() as demo:
 
     )
 
-    ( ################################################ State variables and components for news articles and quiz
+    ( ################################################## State variables and components for news articles and quiz
         
         heading,
         
@@ -71,7 +71,7 @@ with gr.Blocks() as demo:
         
     ) = init
     
-    def hide_news(): ################################# Hide news-related components
+    def hide_news(): ################################### Hide news-related components
         num_headlines = gr.Slider(visible=False)
         category = gr.Radio(visible=False)
         get = gr.Button(visible=False)
@@ -86,28 +86,28 @@ with gr.Blocks() as demo:
 
         return num_headlines, category, get, headline, description, show, summarize, content, ready
 
-    def show_news(): ################################# Show news-related components    
+    def show_news(): ################################### Show news-related components    
         num_headlines = gr.Slider(label="Number of Articles", minimum=1, maximum=10, value=3, step=1, visible=True)
         category = gr.Radio(label="Category (optional)", choices=news.__CATEGORIES__, visible=True)
         get = gr.Button("Get Articles", visible=True)
 
         return num_headlines, category, get
     
-    def show_headline(headlines, descriptions): ###### Show news headlines and descriptions
+    def show_headline(headlines, descriptions): ######## Show news headlines and descriptions
         headline = gr.Radio(label="News Headlines", choices=headlines, value=headlines[0], interactive=True, visible=True)
         description = gr.Textbox(label="Headline Description", value=descriptions[0], visible=True)
         show = gr.Button("Show Content", visible=True)
         
         return headline, description, show
     
-    def show_content(summary): ####################### Show article content and summary
-        summarize = gr.Checkbox(label="Show Summary?", value=True, interactive=True, visible=True)
-        content = gr.Textbox(label="Summary", value=summary, visible=True)
+    def show_content(content): ######################### Show article content and summary
+        summarize = gr.Checkbox(label="Show Summary?", value=False, interactive=True, visible=True)
+        content = gr.Textbox(label="Content", value=content, visible=True)
         ready = gr.Button("Begin Quiz", visible=True)
         
         return summarize, content, ready
     
-    def format_mcq(mcq): ############################# Format multiple choice question for quiz
+    def format_mcq(mcq): ############################### Format multiple choice question for quiz
         if not mcq or not isinstance(mcq, dict):
             print(f"Multiple choice question object is a {type(mcq)} but should be {type(dict())}.")
             return "Invalid multiple choice question.", None
@@ -127,7 +127,7 @@ with gr.Blocks() as demo:
         
         return question, options, answer
     
-    def hide_quiz(): ################################# Hide quiz-related components
+    def hide_quiz(): ################################### Hide quiz-related components
         quiz = [gr.Radio(visible=False) for _ in range(10)]
         submit = gr.Button(visible=False)
         
@@ -136,7 +136,7 @@ with gr.Blocks() as demo:
 
         return read, evaluation, submit, *quiz
 
-    def show_quiz(mcqs): ############################# Show quiz-related components 
+    def show_quiz(mcqs): ############################### Show quiz-related components 
         quiz = [(mcq["question"], mcq["false_answers"], mcq["correct_answer"]) for mcq in mcqs]
         quiz = [(question, random.sample(distractors + [answer], 4), answer) for question, distractors, answer in quiz]
         questions, options, answers = zip(*quiz) if quiz else ([], [], [])
@@ -152,7 +152,7 @@ with gr.Blocks() as demo:
         
         return submit, list(answers), *quiz
     
-    def show_eval(eva): ############################## Show evaluation of user's response to the quiz
+    def show_eval(eva): ################################ Show evaluation of user's response to the quiz
         evaluation = gr.Textbox(label="Evaluation", value=eva, visible=True)
         read = gr.Button("Read Articles", visible=True)
         
@@ -162,7 +162,7 @@ with gr.Blocks() as demo:
     ###### Get and Display News Articles
     ######
     
-    def get_headline(category, num_headlines): ####### Get news headlines based on selected category and number
+    def get_headline(category, num_headlines): ######### Get news headlines based on selected category and number
         articles = news.get_top_headlines(category=category, num_headlines=num_headlines)
         headlines, descriptions = zip(*[(article['title'], article.get('description', 'No description available.')) for article in articles])
         show = show_headline(headlines, descriptions)
@@ -171,7 +171,7 @@ with gr.Blocks() as demo:
     
     get.click(get_headline, inputs=[category, num_headlines], outputs=[articles, descriptions, headline, description, show])
     
-    def get_description(descriptions, headline): ##### Get description for the selected headline
+    def get_description(descriptions, headline): ####### Get description for the selected headline
         description = "No description available."
         
         if   not descriptions:                   print("Descriptions are empty.")
@@ -183,7 +183,7 @@ with gr.Blocks() as demo:
     
     headline.change(get_description, inputs=[descriptions, headline], outputs=[description])
     
-    def get_article(articles, headline): ############# Get article for the selected headline
+    def get_article(articles, headline): ############### Get article for the selected headline
         headlines = [a['title'] for a in articles]
         
         if headline not in headlines: return {}
@@ -191,33 +191,29 @@ with gr.Blocks() as demo:
     
     show.click(get_article, inputs=[articles, headline], outputs=[article])
     
-    def get_content(articles, article): ############## Get content for the selected article
-        if "summary" not in article:
-            idx = articles.index(article)
-            articles[idx]["summary"] = model.get_summary(article, 1)
-            article = articles[idx]
-            
-        return articles, article, *show_content(article.get("summary", "No summary available."))
+    def get_content(article): ########################## Get content for the selected article
+        return article, *show_content(article.get("content", "Content not available."))
     
-    article.change(get_content, inputs=[articles, article], outputs=[articles, article, summarize, content, ready])
+    article.change(get_content, inputs=[article], outputs=[article, summarize, content, ready])
     
-    def toggle_summary(article, summarize): ########## Toggle between showing summary and full content
-        content = "No article available."
-        
+    def toggle_summary(articles, article, summarize): ## Toggle between showing summary and full content
         if not article:                     print("Selected article is empty.")
         elif not isinstance(article, dict): print(f"Selected article is a {type(article)} but should be {type(dict())}.")
-        elif summarize:                     content = article.get("summary", "Summary not available.")
-        else:                               content = article.get("content", "Content not available.")
+        elif not summarize:                 return article.get("content", "Content not available.")
+        elif "summary" not in article:
+                                            idx = articles.index(article)
+                                            articles[idx]["summary"] = model.get_summary(article, 1)
+                                            article = articles[idx]
         
-        return content
+        return  article.get("summary", "Summary not available.")
     
-    summarize.change(toggle_summary, inputs=[article, summarize], outputs=[content])
+    summarize.change(toggle_summary, inputs=[articles, article, summarize], outputs=[content])
     
     ######
     ###### Quiz Generation and Evaluation
     ######
     
-    def get_quiz(content): ########################### Generate quiz questions from the article content
+    def get_quiz(content): ############################# Generate quiz questions from the article content
         multichoicequests = []
         
         if not content:                     mcqs = multichoicequests
@@ -240,7 +236,7 @@ with gr.Blocks() as demo:
     ready.click(get_quiz, inputs=[content], outputs=[
         heading, num_headlines, category, get, headline, description, show, summarize, content, ready, submit, answers, *quiz])
     
-    def get_evaluation(answers, *quiz): ############## Evaluate the user's responses to the quiz
+    def get_evaluation(answers, *quiz): ################ Evaluate the user's responses to the quiz
         results = -1
         
         if not answers:                     print("Answers are empty.")
@@ -262,7 +258,7 @@ with gr.Blocks() as demo:
     
     submit.click(get_evaluation, inputs=[answers, *quiz], outputs=[evaluation, read])
     
-    def read_articles(): ############################# Reset the interface to read articles again
+    def read_articles(): ############################### Reset the interface to read articles again
         return gr.Markdown("## News Articles"), *show_news(), *hide_quiz()
     
     read.click(read_articles, outputs=[heading, num_headlines, category, get, read, evaluation, submit, *quiz])