Upload 4 files

README.md

@@ -1,10 +1,10 @@
 ---
-title: Shugma
-emoji: 📚
-colorFrom: red
-colorTo: yellow
+title: Newnewawea
+emoji: 🚀
+colorFrom: yellow
+colorTo: gray
 sdk: gradio
-sdk_version: 5.43.1
+sdk_version: 5.43.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,400 @@
+import gradio as gr
+import requests
+import json
+import os
+from datetime import datetime
+import base64
+import io
+from PIL import Image
+
+# API URLs from the ECI website
+CAPTCHA_API_URL = "https://gateway-voters.eci.gov.in/api/v1/captcha-service/generateCaptcha/EROLL"
+SEARCH_API_URL = "https://gateway-voters.eci.gov.in/api/v1/elastic/search-by-epic-from-national-display"
+
+# Create directory for saving responses
+RESPONSES_DIR = "."
+if not os.path.exists(RESPONSES_DIR):
+    os.makedirs(RESPONSES_DIR)
+
+def save_json_response(json_data, epic_number, status):
+    """Save JSON response to a file with a timestamp."""
+    timestamp = datetime.now().strftime('%Y%m%d_%H%M%S_%f')[:-3]  # Include milliseconds
+    filename = f"response_{status}_{epic_number}_{timestamp}.json"
+    filepath = os.path.join(RESPONSES_DIR, filename)
+    
+    response_data = {
+        "timestamp": datetime.now().isoformat(),
+        "epic_number": epic_number,
+        "status": status,
+        "response_data": json_data
+    }
+    
+    try:
+        with open(filepath, 'w', encoding='utf-8') as f:
+            json.dump(response_data, f, indent=4, ensure_ascii=False)
+        print(f"Response saved to {filepath}")
+        return filepath
+    except Exception as e:
+        print(f"Error saving JSON response: {e}")
+        return None
+
+def get_new_captcha():
+    """Fetches a new captcha and returns it as a PIL Image object with captcha ID."""
+    try:
+        response = requests.get(CAPTCHA_API_URL, timeout=15)
+        response.raise_for_status()
+        api_data = response.json()
+        
+        base64_image_string = api_data.get('captcha')
+        captcha_id = api_data.get('id')
+        
+        if base64_image_string and captcha_id:
+            # Decode the base64 string into bytes
+            image_bytes = base64.b64decode(base64_image_string)
+            # Create a PIL Image object from the bytes
+            image = Image.open(io.BytesIO(image_bytes))
+            
+            status_msg = f"ᬅ New captcha fetched at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+            return image, captcha_id, status_msg
+        else:
+            return None, None, "Error: Invalid API response format."
+            
+    except requests.exceptions.RequestException as e:
+        return None, None, f"Error: API request failed: {e}"
+    except (ValueError, Exception) as e:
+        return None, None, f"Error: Failed to decode data from API: {e}"
+
+def get_captcha_api():
+    """API-friendly version that returns captcha as base64 for external use."""
+    try:
+        response = requests.get(CAPTCHA_API_URL, timeout=15)
+        response.raise_for_status()
+        api_data = response.json()
+        
+        base64_image_string = api_data.get('captcha')
+        captcha_id = api_data.get('id')
+        
+        if base64_image_string and captcha_id:
+            timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+            return {
+                "success": True,
+                "captcha_id": captcha_id,
+                "captcha_base64": base64_image_string,
+                "timestamp": timestamp,
+                "message": f"Captcha fetched successfully at {timestamp}"
+            }
+        else:
+            return {
+                "success": False,
+                "error": "Invalid API response format",
+                "captcha_id": None,
+                "captcha_base64": None
+            }
+            
+    except Exception as e:
+        return {
+            "success": False,
+            "error": str(e),
+            "captcha_id": None,
+            "captcha_base64": None
+        }
+
+def search_epic_api(epic_number, captcha_data, captcha_id):
+    """API-friendly version of epic search."""
+    if not all([epic_number, captcha_data, captcha_id]):
+        return {
+            "success": False,
+            "error": "Missing required fields (epic_number, captcha_data, captcha_id)",
+            "data": None
+        }
+
+    headers = {
+        'Accept': 'application/json, text/plain, */*',
+        'Content-Type': 'application/json',
+        'Origin': 'https://electoralsearch.eci.gov.in',
+        'Referer': 'https://electoralsearch.eci.gov.in/',
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36',
+    }
+    
+    payload = {
+        "isPortal": True,
+        "epicNumber": epic_number.strip().upper(),
+        "captchaId": captcha_id,
+        "captchaData": captcha_data.strip(),
+        "securityKey": "na"
+    }
+
+    try:
+        response = requests.post(SEARCH_API_URL, headers=headers, json=payload, timeout=30)
+        response_text = response.text.lower()
+        
+        if ("invalid captcha" in response_text or 
+            "captcha is not valid" in response_text):
+            return {
+                "success": False,
+                "error": "Invalid captcha",
+                "captcha_incorrect": True,
+                "data": None,
+                "timestamp": datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+            }
+
+        response.raise_for_status()
+        
+        try:
+            result_json = response.json()
+            # Save successful response
+            saved_file = save_json_response(result_json, epic_number, "success")
+            
+            return {
+                "success": True,
+                "data": result_json,
+                "epic_number": epic_number,
+                "saved_file": saved_file,
+                "timestamp": datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+            }
+            
+        except json.JSONDecodeError:
+            save_json_response({"raw_response": response.text}, epic_number, "non_json")
+            return {
+                "success": False,
+                "error": "Non-JSON response received",
+                "raw_response": response.text,
+                "data": None
+            }
+
+    except Exception as e:
+        return {
+            "success": False,
+            "error": str(e),
+            "data": None
+        }
+
+def perform_epic_search(epic_number, captcha_data, captcha_id):
+    """
+    Submits the EPIC number and captcha data. Returns the JSON response and status.
+    Automatically fetches a new captcha after each attempt.
+    """
+    if not all([epic_number, captcha_data, captcha_id]):
+        return None, "Error: Missing required fields. Please fill all fields.", gr.update(), gr.update()
+
+    headers = {
+        'Accept': 'application/json, text/plain, */*',
+        'Content-Type': 'application/json',
+        'Origin': 'https://electoralsearch.eci.gov.in',
+        'Referer': 'https://electoralsearch.eci.gov.in/',
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36',
+    }
+    
+    payload = {
+        "isPortal": True,
+        "epicNumber": epic_number.strip().upper(),
+        "captchaId": captcha_id,
+        "captchaData": captcha_data.strip(),
+        "securityKey": "na"
+    }
+
+    search_timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+
+    try:
+        response = requests.post(SEARCH_API_URL, headers=headers, json=payload, timeout=30)
+        response_text = response.text.lower()
+        
+        # Check for invalid captcha
+        if ("invalid captcha" in response_text or 
+            "captcha is not valid" in response_text or
+            ("captcha" in response_text and "invalid" in response_text)):
+            
+            save_json_response({
+                "error": "Invalid captcha",
+                "response_text": response.text,
+            }, epic_number, "captcha_failed")
+            
+            status = f"ᭌ Incorrect Captcha at {search_timestamp}. Fetching a new one..."
+            new_image, new_id, captcha_status = get_new_captcha()
+            combined_status = f"{status}\n{captcha_status}"
+            
+            # Return no JSON, update status, and provide the new captcha image and ID
+            return None, combined_status, new_image, new_id
+
+        response.raise_for_status()
+        
+        try:
+            result_json = response.json()
+            saved_file = save_json_response(result_json, epic_number, "success")
+            
+            status = f"ᬅ Search Successful at {search_timestamp}\n"
+            if saved_file:
+                status += f"Response saved to: {os.path.basename(saved_file)}\n"
+            status += "Fetching new captcha for the next search..."
+            
+            new_image, new_id, captcha_status = get_new_captcha()
+            combined_status = f"{status}\n{captcha_status}"
+            
+            return result_json, combined_status, new_image, new_id
+            
+        except json.JSONDecodeError:
+            save_json_response({
+                "raw_response": response.text,
+            }, epic_number, "non_json_response")
+            
+            status = f"᪠ Non-JSON response received at {search_timestamp}. See saved file for details."
+            new_image, new_id, captcha_status = get_new_captcha()
+            combined_status = f"{status}\n{captcha_status}"
+            
+            return {"raw_response": response.text}, combined_status, new_image, new_id
+
+    except requests.exceptions.Timeout:
+        error_msg = f"៱ Request timed out at {search_timestamp}. Please try again."
+        save_json_response({"error": "Timeout"}, epic_number, "timeout")
+        return None, error_msg, gr.update(), gr.update()
+        
+    except requests.exceptions.RequestException as e:
+        status_code = e.response.status_code if e.response else 'N/A'
+        error_details = f"ᭌ Request failed at {search_timestamp}\nStatus: {status_code}\nError: {e}"
+        
+        save_json_response({
+            "error": str(e),
+            "status_code": status_code,
+            "response_text": e.response.text if e.response else ''
+        }, epic_number, "request_failed")
+        
+        return None, error_details, gr.update(), gr.update()
+
+def clear_inputs():
+    """Clear input fields"""
+    return "", ""
+
+# ==============================================================================
+# GRADIO INTERFACE
+# ==============================================================================
+
+with gr.Blocks(theme=gr.themes.Soft(), title="ECI Voter Search") as demo:
+    gr.Markdown("""
+    # 창 ECI Voter Search via EPIC Number
+    
+    **Instructions:**
+    1. A captcha will load automatically.
+    2. Enter an EPIC number and the text from the captcha image.
+    3. Click "Search Voter Details".
+    4. Responses are automatically saved to the `eci_responses` folder.
+    """)
+    
+    captcha_id_state = gr.State()
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### 창 Captcha & Search")
+            
+            with gr.Group():
+                captcha_image = gr.Image(
+                    label="Captcha Image", 
+                    interactive=False, 
+                    type="pil",
+                    height=100
+                )
+                
+                with gr.Row():
+                    fetch_captcha_btn = gr.Button("창 Refresh Captcha", size="sm")
+                    clear_btn = gr.Button("창 Clear Inputs", size="sm")
+            
+            epic_input = gr.Textbox(
+                label="EPIC Number", 
+                placeholder="e.g., ABC1234567"
+            )
+            
+            captcha_input = gr.Textbox(
+                label="Captcha Text",
+                placeholder="Enter the text from the captcha image"
+            )
+            
+            submit_btn = gr.Button("창 Search Voter Details", variant="primary", size="lg")
+
+        with gr.Column(scale=2):
+            gr.Markdown("### 창 Results & Status")
+            
+            status_output = gr.Textbox(
+                label="Status", 
+                interactive=False, 
+                lines=4,
+                value="창 Loading... Fetching initial captcha..."
+            )
+            
+            json_output = gr.JSON(
+                label="Voter Details Response"
+            )
+            
+            gr.Markdown(f"""
+            **창 Auto-Save Location:** All responses are automatically saved in the `{os.path.abspath(RESPONSES_DIR)}` directory.
+            """)
+
+    # Event Handlers
+    
+    # Load initial captcha when the app starts
+    demo.load(
+        fn=get_new_captcha,
+        inputs=None,
+        outputs=[captcha_image, captcha_id_state, status_output]
+    )
+
+    # Handler for the "Refresh Captcha" button
+    fetch_captcha_btn.click(
+        fn=get_new_captcha,
+        inputs=None,
+        outputs=[captcha_image, captcha_id_state, status_output],
+        api_name="get_new_captcha"
+    )
+
+    # Handler for the "Clear Inputs" button
+    clear_btn.click(
+        fn=clear_inputs,
+        inputs=None,
+        outputs=[epic_input, captcha_input]
+    )
+
+    # Handler for the main search button
+    submit_btn.click(
+        fn=perform_epic_search,
+        inputs=[epic_input, captcha_input, captcha_id_state],
+        outputs=[json_output, status_output, captcha_image, captcha_id_state],
+        api_name="search_epic"
+    )
+
+    # Allow submitting with the Enter key from the captcha input field
+    captcha_input.submit(
+        fn=perform_epic_search,
+        inputs=[epic_input, captcha_input, captcha_id_state],
+        outputs=[json_output, status_output, captcha_image, captcha_id_state]
+    )
+
+    # Hidden API interfaces for external access
+    with gr.Group(visible=False):
+        gr.Interface(
+            fn=get_captcha_api,
+            inputs=[],
+            outputs=gr.JSON(),
+            api_name="api_get_captcha"
+        )
+        
+        gr.Interface(
+            fn=search_epic_api,
+            inputs=[
+                gr.Textbox(label="epic_number"),
+                gr.Textbox(label="captcha_data"), 
+                gr.Textbox(label="captcha_id")
+            ],
+            outputs=gr.JSON(),
+            api_name="api_search_epic"
+        )
+
+if __name__ == "__main__":
+    print(f"창 Responses will be saved to: {os.path.abspath(RESPONSES_DIR)}")
+    print("Starting Gradio server...")
+    print("API Endpoints will be available at your server address:")
+    print("  - GET captcha: /api/api_get_captcha")
+    print("  - POST search: /api/api_search_epic")
+    
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        debug=True,
+    )

requirements.txt

@@ -0,0 +1,9 @@
+gradio-client
+requests
+Pillow
+imagehash
+numpy
+tqdm
+transformers
+torch
+huggingface_hub