Update app.py

app.py

@@ -4,15 +4,17 @@ from PIL import Image
 from fastapi import FastAPI
 from starlette.responses import RedirectResponse
 
-# ✅ Patch for Pydantic v2 schema compatibility (important for FastAPI + Gradio)
+# 🔥 Fix for Pydantic v2 compatibility with Gradio
+import gradio.context
 from pydantic import BaseModel
-BaseModel.model_config = {"arbitrary_types_allowed": True}
+if not hasattr(BaseModel, "model_fields"):  # model_fields was renamed from __fields__ in Pydantic v1 → v2
+    BaseModel.model_fields = BaseModel.__fields__
 
-# ✅ Load models
+# 🔁 Load Hugging Face Pipelines
 summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 image_captioner = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
 
-# ✅ Initialize FastAPI app
+# 🚀 Create FastAPI App
 app = FastAPI()
 
 def analyze_input(file, question=None):
@@ -21,13 +23,13 @@ def analyze_input(file, question=None):
 
     filename = file.name.lower()
 
-    # 📷 Image Processing
+    # 🖼️ Image
     if filename.endswith((".png", ".jpg", ".jpeg")):
         image = Image.open(file)
         caption = image_captioner(image)[0]['generated_text']
         return f"📷 Image Interpretation:\n{caption}"
 
-    # 📄 Document Processing
+    # 📄 Document
     elif filename.endswith((".pdf", ".docx", ".pptx", ".xlsx")):
         import pdfplumber
         import docx
@@ -68,22 +70,22 @@ def analyze_input(file, question=None):
     else:
         return "❌ Unsupported file type. Please upload a valid image or document."
 
-# ✅ Gradio Interface
+# 🎛️ Gradio UI
 iface = gr.Interface(
     fn=analyze_input,
     inputs=gr.File(label="Upload Document or Image"),
     outputs=gr.Textbox(label="Result", lines=10),
     title="Document & Image Analysis Web Service",
-    description="Upload a document (PDF, DOCX, PPTX, XLSX) to get a summary or an image to get a caption. Runs fully on CPU."
+    description="Upload a document (PDF, DOCX, PPTX, XLSX) or image to get a summary or caption. CPU-friendly."
 )
 
-# ✅ Wrap in a Tabbed Interface
+# ⌨️ Wrap in Tabbed UI
 demo = gr.TabbedInterface([iface], ["Docs and Images"])
 
-# ✅ Mount to FastAPI app
+# 🔗 Mount Gradio to FastAPI
 app = gr.mount_gradio_app(app, demo, path="/")
 
-# ✅ Redirect base URL to Gradio app
+# 🏠 Base redirect
 @app.get("/")
 def home():
     return RedirectResponse(url="/")