File size: 3,000 Bytes
2380f6f |
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 |
"""Topic extraction endpoints"""
from fastapi import APIRouter, HTTPException
from datetime import datetime
import logging
from services.topic_service import topic_service
from models.topic import (
TopicRequest,
TopicResponse,
BatchTopicRequest,
BatchTopicResponse,
)
router = APIRouter()
logger = logging.getLogger(__name__)
@router.post("/extract", response_model=TopicResponse, tags=["Topic Extraction"])
async def extract_topic(request: TopicRequest):
"""
Extract a topic from a given text/argument
- **text**: The input text or argument to extract topic from (5-5000 chars)
Returns the extracted topic description
"""
try:
# Extract topic
topic = topic_service.extract_topic(request.text)
# Build response
response = TopicResponse(
text=request.text,
topic=topic,
timestamp=datetime.now().isoformat()
)
logger.info(f"Topic extracted: {topic[:50]}...")
return response
except ValueError as e:
logger.error(f"Validation error: {str(e)}")
raise HTTPException(status_code=400, detail=str(e))
except Exception as e:
logger.error(f"Topic extraction error: {str(e)}")
raise HTTPException(status_code=500, detail=f"Topic extraction failed: {str(e)}")
@router.post("/batch-extract", response_model=BatchTopicResponse, tags=["Topic Extraction"])
async def batch_extract_topics(request: BatchTopicRequest):
"""
Extract topics from multiple texts/arguments
- **texts**: List of texts to extract topics from (max 50)
Returns extracted topics for all texts
"""
try:
# Batch extract topics
topics = topic_service.batch_extract_topics(request.texts)
# Build response
results = []
timestamp = datetime.now().isoformat()
for i, text in enumerate(request.texts):
if topics[i] is not None:
results.append(
TopicResponse(
text=text,
topic=topics[i],
timestamp=timestamp
)
)
else:
# Skip failed extractions or handle as needed
logger.warning(f"Failed to extract topic for text at index {i}")
logger.info(f"Batch topic extraction completed: {len(results)}/{len(request.texts)} successful")
return BatchTopicResponse(
results=results,
total_processed=len(results),
timestamp=timestamp
)
except ValueError as e:
logger.error(f"Validation error: {str(e)}")
raise HTTPException(status_code=400, detail=str(e))
except Exception as e:
logger.error(f"Batch topic extraction error: {str(e)}")
raise HTTPException(status_code=500, detail=f"Batch topic extraction failed: {str(e)}")
|