Update app.py

app.py

@@ -1,11 +1,16 @@
 import gradio as gr
-import groq_gradio
+from groq import Groq
 import os
 from PIL import Image, ImageDraw, ImageFont
 from datetime import datetime
 import json
 import tempfile
 
+# Initialize Groq client
+client = Groq(
+    api_key=os.getenv("GROQ_API_KEY")
+)
+
 class QuizApp:
     def __init__(self):
         self.current_questions = []
@@ -27,27 +32,34 @@ class QuizApp:
         }}
         Only return the JSON, no other text."""
         
-        # Use groq-gradio's built-in interface for LLM
-        interface = gr.load(
-            name='llama2-70b-4096',
-            src=groq_gradio.registry,
+        # Use Groq API directly
+        response = client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model="llama2-70b-4096",
+            temperature=0.7,
+            max_tokens=1024
         )
         
-        response = interface.predict(prompt)
-        questions = json.loads(response)
-        self.current_questions = questions["questions"]
-        return json.dumps(questions["questions"], indent=2)
+        try:
+            questions = json.loads(response.choices[0].message.content)
+            self.current_questions = questions["questions"]
+            return json.dumps(questions["questions"], indent=2)
+        except json.JSONDecodeError:
+            return json.dumps({"error": "Failed to generate valid questions. Please try again."})
     
     def calculate_score(self, answers):
-        answers = json.loads(answers)
-        total = len(self.current_questions)
-        correct = 0
-        
-        for q, a in zip(self.current_questions, answers):
-            if set(a) == set(q["correct_answers"]):
-                correct += 1
-                
-        return (correct / total) * 100
+        try:
+            answers = json.loads(answers)
+            total = len(self.current_questions)
+            correct = 0
+            
+            for q, a in zip(self.current_questions, answers):
+                if set(a) == set(q["correct_answers"]):
+                    correct += 1
+                    
+            return (correct / total) * 100
+        except:
+            return 0
     
     def generate_certificate(self, name, score, course_name, company_logo=None, participant_photo=None):
         # Create certificate
@@ -77,15 +89,21 @@ class QuizApp:
         
         # Add logo if provided
         if company_logo is not None:
-            logo = Image.open(company_logo)
-            logo = logo.resize((150, 150))
-            certificate.paste(logo, (50, 50))
+            try:
+                logo = Image.open(company_logo)
+                logo = logo.resize((150, 150))
+                certificate.paste(logo, (50, 50))
+            except Exception as e:
+                print(f"Error adding logo: {e}")
         
         # Add photo if provided
         if participant_photo is not None:
-            photo = Image.open(participant_photo)
-            photo = photo.resize((150, 150))
-            certificate.paste(photo, (1000, 50))
+            try:
+                photo = Image.open(participant_photo)
+                photo = photo.resize((150, 150))
+                certificate.paste(photo, (1000, 50))
+            except Exception as e:
+                print(f"Error adding photo: {e}")
         
         # Save to temporary file
         temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
@@ -95,34 +113,59 @@ class QuizApp:
 def create_quiz_app():
     quiz_app = QuizApp()
     
-    with gr.Blocks(title="Interactive Quiz Generator") as demo:
-        gr.Markdown("# Interactive Quiz Generator")
+    with gr.Blocks(title="QuizForge AI") as demo:
+        gr.Markdown("# QuizForge AI")
+        gr.Markdown("### Generate personalized quizzes and earn certificates powered by AI")
         
         # User Information Tab
         with gr.Tab("Step 1: User Information"):
-            name = gr.Textbox(label="Full Name", placeholder="Enter your full name")
-            email = gr.Textbox(label="Email", placeholder="Enter your email")
-            text_input = gr.Textbox(label="Content for Quiz", 
-                                  placeholder="Enter the text content for generating questions",
-                                  lines=10)
-            num_questions = gr.Slider(minimum=1, maximum=10, value=5, step=1,
-                                    label="Number of Questions")
-            company_logo = gr.Image(label="Company Logo (Optional)", type="filepath")
-            participant_photo = gr.Image(label="Participant Photo (Optional)", 
-                                      type="filepath")
-            generate_btn = gr.Button("Generate Quiz")
+            with gr.Row():
+                name = gr.Textbox(label="Full Name", placeholder="Enter your full name")
+                email = gr.Textbox(label="Email", placeholder="Enter your email")
+            
+            text_input = gr.Textbox(
+                label="Content for Quiz", 
+                placeholder="Enter the text content for generating questions",
+                lines=10
+            )
+            
+            with gr.Row():
+                num_questions = gr.Slider(
+                    minimum=1, 
+                    maximum=10, 
+                    value=5, 
+                    step=1,
+                    label="Number of Questions"
+                )
+                
+            with gr.Row():
+                company_logo = gr.Image(label="Company Logo (Optional)", type="filepath")
+                participant_photo = gr.Image(label="Participant Photo (Optional)", type="filepath")
+                
+            generate_btn = gr.Button("Generate Quiz", variant="primary")
         
         # Quiz Tab
         with gr.Tab("Step 2: Take Quiz"):
+            gr.Markdown("""
+            ### Instructions:
+            1. Questions will appear in JSON format below
+            2. Each question has multiple options
+            3. Enter answers as array indices (e.g., [0, 2] for first and third options)
+            4. Submit when ready to get your score
+            """)
+            
             questions_display = gr.JSON(label="Questions")
             answers_input = gr.JSON(label="Your Answers (Enter indices of correct options)")
-            submit_btn = gr.Button("Submit Answers")
+            submit_btn = gr.Button("Submit Answers", variant="primary")
             score_display = gr.Number(label="Your Score")
         
         # Certificate Tab
         with gr.Tab("Step 3: Get Certificate"):
-            course_name = gr.Textbox(label="Course Name", 
-                                   value="Interactive Quiz Course")
+            gr.Markdown("### Your certificate will be generated automatically if you score 80% or above")
+            course_name = gr.Textbox(
+                label="Course Name", 
+                value="QuizForge AI Certification"
+            )
             certificate_display = gr.Image(label="Certificate")
         
         # Event handlers
@@ -150,5 +193,9 @@ def create_quiz_app():
     return demo
 
 if __name__ == "__main__":
+    if not os.getenv("GROQ_API_KEY"):
+        print("Please set your GROQ_API_KEY environment variable")
+        exit(1)
+        
     demo = create_quiz_app()
-    demo.launch()
+    demo.launch(share=True)