Spaces:
Running
Running
File size: 8,317 Bytes
18b8b90 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 | """
Comprehensive OCR Provider Diagnostic Script
Tests each provider individually to identify exact failure reasons.
"""
import asyncio
import sys
import os
import traceback
# Ensure imports work
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from PIL import Image
# Create a small test image (white square with some text-like pattern)
def create_test_image():
"""Create a minimal test image for OCR testing."""
img = Image.new("RGB", (200, 100), color=(255, 255, 255))
# Draw some dark pixels to simulate text
for x in range(20, 180):
for y in range(40, 60):
if (x + y) % 3 == 0:
img.putpixel((x, y), (0, 0, 0))
return img
def separator(title: str):
print(f"\n{'='*60}")
print(f" {title}")
print(f"{'='*60}")
async def test_groq():
"""Test Groq provider."""
separator("TESTING GROQ PROVIDER")
try:
from app.core.config import settings
print(f" GROQ_API_KEY present: {bool(settings.GROQ_API_KEY)}")
print(f" GROQ_API_KEY prefix: {settings.GROQ_API_KEY[:10]}..." if settings.GROQ_API_KEY else " NO KEY")
from groq import AsyncGroq
print(" ✅ groq package imported successfully")
client = AsyncGroq(api_key=settings.GROQ_API_KEY)
print(" ✅ AsyncGroq client created")
# Quick test: just list models to verify API key works
models = await client.models.list()
vision_models = [m.id for m in models.data if "vision" in m.id.lower()]
print(f" ✅ API key valid! Available vision models: {vision_models}")
# Now test actual OCR
from app.services.ocr.groq_provider import GroqOCRProvider
from app.schemas import DocumentType
provider = GroqOCRProvider()
print(f" ✅ GroqOCRProvider instantiated, available: {provider.is_available()}")
img = create_test_image()
prompt = "Extract text from this image. Return JSON: {\"text\": \"...\"}"
result = await provider.extract(img, DocumentType.ID_CARD, prompt)
print(f" ✅ OCR SUCCESS! Result: {result}")
return True
except Exception as e:
print(f" ❌ FAILED: {type(e).__name__}: {e}")
traceback.print_exc()
return False
async def test_gemini():
"""Test Gemini provider."""
separator("TESTING GEMINI PROVIDER")
try:
from app.core.config import settings
print(f" GEMINI_API_KEY present: {bool(settings.GEMINI_API_KEY)}")
print(f" GEMINI_API_KEY prefix: {settings.GEMINI_API_KEY[:10]}..." if settings.GEMINI_API_KEY else " NO KEY")
from google import genai
print(" ✅ google-genai package imported successfully")
client = genai.Client(api_key=settings.GEMINI_API_KEY)
print(" ✅ genai.Client created")
# Quick model list test
# Test actual OCR
from app.services.ocr.gemini_provider import GeminiOCRProvider
from app.schemas import DocumentType
provider = GeminiOCRProvider()
print(f" ✅ GeminiOCRProvider instantiated, available: {provider.is_available()}")
img = create_test_image()
prompt = "Extract text from this image. Return JSON: {\"text\": \"...\"}"
result = await provider.extract(img, DocumentType.ID_CARD, prompt)
print(f" ✅ OCR SUCCESS! Result: {result}")
return True
except Exception as e:
print(f" ❌ FAILED: {type(e).__name__}: {e}")
traceback.print_exc()
return False
async def test_openai():
"""Test OpenAI provider."""
separator("TESTING OPENAI PROVIDER")
try:
from app.core.config import settings
print(f" OPENAI_API_KEY present: {bool(settings.OPENAI_API_KEY)}")
print(f" OPENAI_API_KEY prefix: {settings.OPENAI_API_KEY[:15]}..." if settings.OPENAI_API_KEY else " NO KEY")
from openai import AsyncOpenAI
print(" ✅ openai package imported successfully")
client = AsyncOpenAI(api_key=settings.OPENAI_API_KEY)
print(" ✅ AsyncOpenAI client created")
# Test actual OCR
from app.services.ocr.openai_provider import OpenAIOCRProvider
from app.schemas import DocumentType
provider = OpenAIOCRProvider()
print(f" ✅ OpenAIOCRProvider instantiated, available: {provider.is_available()}")
img = create_test_image()
prompt = "Extract text from this image. Return JSON: {\"text\": \"...\"}"
result = await provider.extract(img, DocumentType.ID_CARD, prompt)
print(f" ✅ OCR SUCCESS! Result: {result}")
return True
except Exception as e:
print(f" ❌ FAILED: {type(e).__name__}: {e}")
traceback.print_exc()
return False
async def test_huggingface():
"""Test HuggingFace provider."""
separator("TESTING HUGGINGFACE PROVIDER")
try:
from app.core.config import settings
print(f" HUGGINGFACE_API_KEY present: {bool(settings.HUGGINGFACE_API_KEY)}")
print(f" HUGGINGFACE_API_KEY prefix: {settings.HUGGINGFACE_API_KEY[:10]}..." if settings.HUGGINGFACE_API_KEY else " NO KEY")
from huggingface_hub import InferenceClient
print(" ✅ huggingface_hub package imported successfully")
from app.services.ocr.huggingface_provider import _HF_MODEL
print(f" HF Model: {_HF_MODEL}")
client = InferenceClient(provider="hf-inference", api_key=settings.HUGGINGFACE_API_KEY)
print(" ✅ InferenceClient created (hf-inference provider)")
# Test actual OCR
from app.services.ocr.huggingface_provider import HuggingFaceOCRProvider
from app.schemas import DocumentType
provider = HuggingFaceOCRProvider()
print(f" ✅ HuggingFaceOCRProvider instantiated, available: {provider.is_available()}")
img = create_test_image()
prompt = "Extract text from this image. Return JSON: {\"text\": \"...\"}"
result = await provider.extract(img, DocumentType.ID_CARD, prompt)
print(f" ✅ OCR SUCCESS! Result: {result}")
return True
except Exception as e:
print(f" ❌ FAILED: {type(e).__name__}: {e}")
traceback.print_exc()
return False
async def main():
separator("OCR PROVIDER DIAGNOSTIC")
print(f" Python: {sys.version}")
print(f" CWD: {os.getcwd()}")
# Check installed packages
print("\n Checking installed packages...")
packages = ["groq", "openai", "google.generativeai", "google.genai", "PIL", "json_repair", "pydantic_settings"]
for pkg in packages:
try:
__import__(pkg)
print(f" ✅ {pkg}")
except ImportError:
print(f" ❌ {pkg} NOT INSTALLED")
# Check settings
print("\n Checking settings...")
try:
from app.core.config import settings
print(f" GROQ_API_KEY: {'✅ set' if settings.GROQ_API_KEY else '❌ empty'}")
print(f" GEMINI_API_KEY: {'✅ set' if settings.GEMINI_API_KEY else '❌ empty'}")
print(f" OPENAI_API_KEY: {'✅ set' if settings.OPENAI_API_KEY else '❌ empty'}")
print(f" HUGGINGFACE_API_KEY:{'✅ set' if settings.HUGGINGFACE_API_KEY else '❌ empty'}")
except Exception as e:
print(f" ❌ Settings load failed: {e}")
traceback.print_exc()
return
# Test each provider
results = {}
results["groq"] = await test_groq()
results["gemini"] = await test_gemini()
results["openai"] = await test_openai()
results["huggingface"] = await test_huggingface()
# Summary
separator("SUMMARY")
for name, success in results.items():
status = "✅ WORKING" if success else "❌ BROKEN"
print(f" {name:15s} → {status}")
working = sum(1 for v in results.values() if v)
print(f"\n {working}/4 providers working")
if working == 0:
print(" ⚠️ NO PROVIDERS WORKING — OCR service will fail!")
sys.stdout.flush()
sys.stderr.flush()
if __name__ == "__main__":
asyncio.run(main())
|