Update main.py

main.py

@@ -1,15 +1,24 @@
 # main.py
 from fastapi import FastAPI
-from pydantic import BaseModel, Field, HttpUrl
-from typing import List
+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):
-    text: str
-    url: HttpUrl  # Pydantic validates this is a valid URL
-    image: str    # Expecting a base64 encoded string
+    # 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):
@@ -39,7 +48,8 @@ app = FastAPI(
 # 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",                  # Your local development URL
+    "http://localhost:3000", 
+    "https://6000-firebase-studio-1758293511500.cluster-6dx7corvpngoivimwvvljgokdw.cloudworkstations.dev/"
 ]
 
 app.add_middleware(
@@ -57,22 +67,24 @@ async def check_request_format(request: RequestPacket):
     This endpoint receives a request packet, validates its structure,
     and returns a mock analysis response.
     """
-    # For your debugging: print a confirmation to the server console
     print("✅ Request received and successfully validated!")
-    print(f"Received URL: {request.url}")
-    print(f"Text length: {len(request.text)}")
-    print(f"Image string starts with: {request.image[:30]}...")
+    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]}...")
 
-    # -- 4. Create and Return the Mock Response --
+    # The mock response logic remains the same
     mock_response = ResponsePacket(
-        summary="This is a simulated summary based on the provided text. The analysis suggests the content requires further verification due to conflicting reports from primary sources.",
+        summary="This is a simulated summary for the flexible input provided.",
         analysis=Analysis(
             isMisinformation=True,
-            reasoning="The claim contradicts established facts from reputable news organizations and scientific consensus. The provided image appears to be digitally altered.",
+            reasoning="The claim contradicts established facts.",
             confidenceScore=0.88
         ),
         sources=[
-            Source(name="Verified Source A", description="Provides data that conflicts with the user's text."),
+            Source(name="Verified Source A", description="Provides conflicting data."),
             Source(name="FactCheck Organization B", description="Previously debunked a similar claim.")
         ]
     )