Update app.py

app.py

@@ -23,8 +23,6 @@ def system_instructions(context):
 def generate_quiz(context):
     formatted_prompt = system_instructions(context)
 
-    pre_prompt = [{"role": "system", "content": formatted_prompt}]
-
     generate_kwargs = dict(
             temperature=0.1,
             max_new_tokens=2048,
@@ -33,14 +31,31 @@ def generate_quiz(context):
             do_sample=True,
             seed=42,)
         
-    try:
-        response = client.text_generation(
-            formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False,
-        )
-        return response
+    retries = 0
+    response = None
+    
+    while retries < MAX_RETRIES:
+        retries += 1
+        try:
+            response = client.text_generation(
+                formatted_prompt,
+                **generate_kwargs,
+                stream=False,
+                details=False,
+                return_full_text=False,
+            )
+            break
         
-    except Exception as e:
-        return {"error": str(e)}
+        except (RequestException, SystemExit) as e:
+            if retries == MAX_RETRIES:
+                return {"error": f"Failed after {MAX_RETRIES} retries. Error: {str(e)}"}
+            
+            time.sleep(retries)
+    
+    if response is None:
+        return {"error": "Max retries exceeded without success."}
+
+    return response
 
 @app.route("/", methods=["GET", "POST"])
 def generate_quiz_page():