FastAPI-Test / main.py
PercivalFletcher's picture
Update main.py
c56dc42 verified
raw
history blame
3.26 kB
# main.py
from fastapi import FastAPI
from pydantic import BaseModel, Field, HttpUrl,root_validator, validator
from typing import List, Optional
from fastapi.middleware.cors import CORSMiddleware
# -- 1. Define Pydantic Models for Data Validation --
# This model defines the structure of the incoming request JSON
class RequestPacket(BaseModel):
# 1. Make fields optional by using Optional[<type>] and setting default to None
text: Optional[str] = None
url: Optional[HttpUrl] = None
image: Optional[str] = None
# 2. Add a validator to ensure at least one field is provided
@root_validator(pre=False, skip_on_failure=True)
def check_at_least_one_field_present(cls, values):
"""Ensure at least one of text, url, or image is provided."""
if not any(values.values()):
raise ValueError('At least one of the fields (text, url, image) must be provided.')
return values
# These models define the structure of the outgoing response JSON
class Analysis(BaseModel):
isMisinformation: bool
reasoning: str
confidenceScore: float = Field(
...,
ge=0, # Must be greater than or equal to 0
le=1 # Must be less than or equal to 1
)
class Source(BaseModel):
name: str
description: str
class ResponsePacket(BaseModel):
summary: str
analysis: Analysis
sources: List[Source]
# -- 2. Create the FastAPI Application --
app = FastAPI(
title="Backend Checker API",
description="A mock API to validate requests and send simulated responses."
)
# ⬇️ ADD THIS MIDDLEWARE SECTION
# This tells the API to accept requests from your frontend
origins = [
"https://your-firebase-app-name.web.app", # Your production frontend URL
"http://localhost:3000",
"https://6000-firebase-studio-1758293511500.cluster-6dx7corvpngoivimwvvljgokdw.cloudworkstations.dev/"
]
app.add_middleware(
CORSMiddleware,
allow_origins=origins,
allow_credentials=True,
allow_methods=["*"], # Allows all methods, including POST
allow_headers=["*"], # Allows all headers
)
# -- 3. Define the API Endpoint --
@app.post("/check", response_model=ResponsePacket)
async def check_request_format(request: RequestPacket):
"""
This endpoint receives a request packet, validates its structure,
and returns a mock analysis response.
"""
print("✅ Request received and successfully validated!")
if request.text:
print(f"Text received (length): {len(request.text)}")
if request.url:
print(f"URL received: {request.url}")
if request.image:
print(f"Image data received, starts with: {request.image[:30]}...")
# The mock response logic remains the same
mock_response = ResponsePacket(
summary="This is a simulated summary for the flexible input provided.",
analysis=Analysis(
isMisinformation=True,
reasoning="The claim contradicts established facts.",
confidenceScore=0.88
),
sources=[
Source(name="Verified Source A", description="Provides conflicting data."),
Source(name="FactCheck Organization B", description="Previously debunked a similar claim.")
]
)
return mock_response