Update app.py

app.py

@@ -407,6 +407,13 @@ groq_api_key = os.environ.get("GROQ_API_KEY")
 if not isinstance(groq_api_key, str):
     raise EnvironmentError("GROQ_API_KEY not found. Please add it to Hugging Face Space secrets.")
 os.environ["GROQ_API_KEY"] = groq_api_key
+class StructuredOutput(BaseModel):
+    doc_type_1: Optional[str] = Field(None)
+    confidence_1: Optional[float] = Field(None)
+    doc_type_2: Optional[str] = Field(None)
+    confidence_2: Optional[float] = Field(None)
+    doc_type_3: Optional[str] = Field(None)
+    confidence_3: Optional[float] = Field(None)
 
 
 phi_agent = Agent(
@@ -423,13 +430,6 @@ phi_agent = Agent(
     show_tool_calls=True,
     monitoring=True
 )
-class StructuredOutput(BaseModel):
-    doc_type_1: Optional[str] = Field(None)
-    confidence_1: Optional[float] = Field(None)
-    doc_type_2: Optional[str] = Field(None)
-    confidence_2: Optional[float] = Field(None)
-    doc_type_3: Optional[str] = Field(None)
-    confidence_3: Optional[float] = Field(None)
 
 def extract_text_from_pdf(file_path):
     try: