Spaces:

Vjay15
/

Quiz_Solver

Sleeping

App Files Files Community

Vjay15 commited on Dec 12, 2025

Commit

78331be

verified ·

1 Parent(s): c2ae850

Upload 3 files

Browse files

Files changed (2) hide show

main.py +1 -76
solver.py +39 -4

main.py CHANGED Viewed

@@ -6,7 +6,7 @@ from solver import solve_quiz
 import logging
 # Configure logging
-logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 load_dotenv()
@@ -20,81 +20,6 @@ class QuizRequest(BaseModel):
     class Config:
         extra = "allow"
-from fastapi.responses import HTMLResponse
-@app.get("/", response_class=HTMLResponse)
-async def home():
-    return """
-    <!DOCTYPE html>
-    <html>
-    <head>
-        <title>Quiz Solver</title>
-        <style>
-            body { font-family: sans-serif; max-width: 600px; margin: 40px auto; padding: 20px; }
-            .form-group { margin-bottom: 15px; }
-            label { display: block; margin-bottom: 5px; }
-            input { width: 100%; padding: 8px; box-sizing: border-box; }
-            button { padding: 10px 20px; background: #007bff; color: white; border: none; cursor: pointer; }
-            button:hover { background: #0056b3; }
-            #status { margin-top: 20px; padding: 10px; display: none; }
-            .success { background: #d4edda; color: #155724; }
-            .error { background: #f8d7da; color: #721c24; }
-        </style>
-    </head>
-    <body>
-        <h1>Quiz Solver Agent</h1>
-        <div class="form-group">
-            <label>Email:</label>
-            <input type="email" id="email" value="22f3000730@ds.iitm.ac.in">
-        </div>
-        <div class="form-group">
-            <label>Secret:</label>
-            <input type="password" id="secret" placeholder="Enter your secret">
-        </div>
-        <div class="form-group">
-            <label>Start URL:</label>
-            <input type="text" id="url" value="https://tds-llm-analysis.s-anand.net/project2">
-        </div>
-        <button onclick="startSolver()">Start Solving</button>
-        <div id="status"></div>
-        <script>
-            async function startSolver() {
-                const email = document.getElementById('email').value;
-                const secret = document.getElementById('secret').value;
-                const url = document.getElementById('url').value;
-                const statusDiv = document.getElementById('status');
-                statusDiv.style.display = 'block';
-                statusDiv.className = '';
-                statusDiv.textContent = 'Sending request...';
-                try {
-                    const response = await fetch('/', {
-                        method: 'POST',
-                        headers: { 'Content-Type': 'application/json' },
-                        body: JSON.stringify({ email, secret, url })
-                    });
-                    const data = await response.json();
-                    if (response.ok) {
-                        statusDiv.className = 'success';
-                        statusDiv.textContent = 'Success: ' + data.message;
-                    } else {
-                        statusDiv.className = 'error';
-                        statusDiv.textContent = 'Error: ' + (data.detail || 'Unknown error');
-                    }
-                } catch (e) {
-                    statusDiv.className = 'error';
-                    statusDiv.textContent = 'Network Error: ' + e.message;
-                }
-            }
-        </script>
-    </body>
-    </html>
-    """
 @app.post("/", status_code=200)
 async def solve_quiz_endpoint(request: QuizRequest, background_tasks: BackgroundTasks):
     logger.info(f"Received quiz request for URL: {request.url}")

 import logging
 # Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 load_dotenv()
     class Config:
         extra = "allow"
 @app.post("/", status_code=200)
 async def solve_quiz_endpoint(request: QuizRequest, background_tasks: BackgroundTasks):
     logger.info(f"Received quiz request for URL: {request.url}")

solver.py CHANGED Viewed

@@ -10,9 +10,14 @@ from agent import get_agent
 logger = logging.getLogger(__name__)
 async def solve_quiz(initial_url: str, email: str, secret: str):
-    logger.info(f"Starting quiz solver workflow for {email}")
     current_url = initial_url
     async with async_playwright() as p:
@@ -22,10 +27,24 @@ async def solve_quiz(initial_url: str, email: str, secret: str):
         try:
             while current_url:
                 # Generate a NEW session ID for each task/URL to keep memory clean
                 import uuid
                 session_id = str(uuid.uuid4())
-                logger.info(f"Started new agent session for {current_url}: {session_id}")
                 logger.info(f"Navigating to {current_url}")
                 await page.goto(current_url)
@@ -118,8 +137,20 @@ async def solve_quiz(initial_url: str, email: str, secret: str):
                 # Use agent to solve (initialize here if needed, but we use get_agent() outside if we wanted persistent agent object,
                 # but we want fresh memory per task, so we rely on session_id)
                 agent = get_agent()
                 prompt = f"""
 You are a highly capable Quiz Solver Agent.
 Current Page URL: {current_url}
@@ -134,6 +165,7 @@ Solve the task on the current page.
 **GUIDELINES**
 - **Conciseness**: Plan and explain in **2-3 lines maximum**.
 - **Action**: Respond **IMMEDIATELY** with a tool call or the final JSON. **DO NOT** output conversational text or plans like "I need to...". Just run the code.
 **TOOL USAGE**
 - **Secret Codes**: Return exactly as requested (no extra spaces).
@@ -263,6 +295,7 @@ Now return the final JSON for submission:
                                 logger.info("Answer correct! Moving to next URL.")
                             current_url = next_url
                             break # Break retry loop to process new URL
                         # No new URL provided
@@ -272,8 +305,10 @@ Now return the final JSON for submission:
                             break # Break retry loop
                         else:
                             logger.warning(f"Answer incorrect: {submission_response.get('reason')}")
-                            logger.info("No new URL provided. Retrying same URL in 2 seconds...")
-                            await asyncio.sleep(2)
                             # Break inner loop to refresh page and try again
                             break

 logger = logging.getLogger(__name__)
 async def solve_quiz(initial_url: str, email: str, secret: str):
+    import uuid
+    task_id = str(uuid.uuid4())[:8]
+    logger.info(f"[{task_id}] Starting quiz solver workflow for {email}")
     current_url = initial_url
+    last_url = None
+    url_attempts = 0
+    failure_details = None
     async with async_playwright() as p:
         try:
             while current_url:
+                # Generate a NEW session ID for each task/URL to keep memory clean
+                import uuid
                 # Generate a NEW session ID for each task/URL to keep memory clean
                 import uuid
                 session_id = str(uuid.uuid4())
+                # Update attempt counter
+                if current_url == last_url:
+                    url_attempts += 1
+                else:
+                    last_url = current_url
+                    url_attempts = 1
+                if url_attempts > 60:
+                    logger.error(f"Max attempts (60) reached for {current_url}. Stopping loop.")
+                    break
+                logger.info(f"Started new agent session for {current_url} (Attempt {url_attempts}): {session_id}")
                 logger.info(f"Navigating to {current_url}")
                 await page.goto(current_url)
                 # Use agent to solve (initialize here if needed, but we use get_agent() outside if we wanted persistent agent object,
                 # but we want fresh memory per task, so we rely on session_id)
                 agent = get_agent()
+                failure_info = ""
+                if failure_details:
+                    failure_info = f"""
+***PREVIOUS FAILED SUBMISSION***
+- The previous answer you submitted was INCORRECT.
+- **Submitted Answer**: {json.dumps(failure_details['answer'].get('answer', 'UNKNOWN'))}
+- **Error/Reason**: {failure_details['reason']}
+- **GUIDANCE**: Analyze the error message carefully. Do NOT repeat the same answer. Try a different approach or format.
+***
+"""
                 prompt = f"""
+{failure_info}
 You are a highly capable Quiz Solver Agent.
 Current Page URL: {current_url}
 **GUIDELINES**
 - **Conciseness**: Plan and explain in **2-3 lines maximum**.
 - **Action**: Respond **IMMEDIATELY** with a tool call or the final JSON. **DO NOT** output conversational text or plans like "I need to...". Just run the code.
+- Have a dummy answer for the intial request
 **TOOL USAGE**
 - **Secret Codes**: Return exactly as requested (no extra spaces).
                                 logger.info("Answer correct! Moving to next URL.")
                             current_url = next_url
+                            failure_details = None # Reset on success
                             break # Break retry loop to process new URL
                         # No new URL provided
                             break # Break retry loop
                         else:
                             logger.warning(f"Answer incorrect: {submission_response.get('reason')}")
+                            logger.info("No new URL provided. Retrying same URL in 20 seconds...")
+                            current_url = current_url # Just to be explicit, logic loops anyway
+                            failure_details = {"answer": answer_payload, "reason": submission_response.get("reason")}
+                            await asyncio.sleep(20)
                             # Break inner loop to refresh page and try again
                             break