Add files using upload-large-folder tool

.idea/workspace.xml

@@ -71,7 +71,7 @@
       <window_info id="Commit_Guest" show_stripe_button="false" />
       <window_info id="Pull Requests" />
       <window_info id="Learn" />
-      <window_info active="true" content_ui="combo" id="Project" order="0" visible="true" weight="0.16933593" />
+      <window_info active="true" content_ui="combo" id="Project" order="0" visible="true" weight="0.28670636" />
       <window_info id="Commit" order="1" weight="0.25" />
       <window_info id="Structure" order="2" side_tool="true" weight="0.25" />
       <window_info anchor="bottom" id="Database Changes" />
@@ -81,7 +81,7 @@
       <window_info anchor="bottom" id="Version Control" order="0" />
       <window_info anchor="bottom" id="Problems" order="1" />
       <window_info anchor="bottom" id="Problems View" order="2" />
-      <window_info active="true" anchor="bottom" id="Terminal" order="3" visible="true" weight="0.3795139" />
+      <window_info active="true" anchor="bottom" id="Terminal" order="3" visible="true" weight="0.27142859" />
       <window_info anchor="bottom" id="Services" order="4" />
       <window_info anchor="bottom" id="Python Packages" order="5" weight="0.1" />
       <window_info anchor="bottom" id="Debug" order="6" weight="0.29618055" />
@@ -97,7 +97,7 @@
       <window_info anchor="right" id="Database" order="2" weight="0.25" />
       <window_info anchor="right" id="Gradle" order="3" weight="0.25" />
       <window_info anchor="right" id="Maven" order="4" weight="0.25" />
-      <window_info anchor="right" id="CodeGPT" order="5" weight="0.30566406" />
+      <window_info active="true" anchor="right" id="CodeGPT" order="5" visible="true" weight="0.32969576" />
       <window_info anchor="right" id="Plots" order="6" weight="0.1" />
     </layout>
   </component>
@@ -187,7 +187,7 @@
       <workItem from="1729970270429" duration="53000" />
       <workItem from="1729970419018" duration="9867000" />
       <workItem from="1730030408588" duration="2251000" />
-      <workItem from="1730037237796" duration="29429000" />
+      <workItem from="1730037237796" duration="32282000" />
     </task>
     <servers />
   </component>

fastapi_app.py

@@ -0,0 +1,57 @@
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from typing import List
+import uvicorn
+from medimageinsightmodel import MedImageInsight
+import base64
+
+# Initialize FastAPI app
+app = FastAPI(title="Medical Image Analysis API")
+
+# Initialize model
+classifier = MedImageInsight(
+    model_dir="2024.09.27",
+    vision_model_name="medimageinsigt-v1.0.0.pt",
+    language_model_name="language_model.pth"
+)
+classifier.load_model()
+
+
+class ClassificationRequest(BaseModel):
+    images: List[str]  # Base64 encoded images
+    labels: List[str]
+    multilabel : bool = False
+
+class EmbeddingRequest(BaseModel):
+    images: List[str] = None  # Base64 encoded images
+    texts: List[str] = None
+
+@app.post("/predict")
+async def predict(request: ClassificationRequest):
+    try:
+        results = classifier.predict(
+            images=request.images,
+            labels=request.labels,
+            multilabel = request.multilabel
+        )
+        return {"predictions": results}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/encode")
+async def encode(request: EmbeddingRequest):
+    try:
+        results = classifier.encode(images=request.images, texts= request.texts)
+        results["image_embeddings"] = results["image_embeddings"].tolist() if results["image_embeddings"] is not None else None
+        results["text_embeddings"] = results["text_embeddings"].tolist() if results["text_embeddings"] is not None else None
+
+        return results
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/health")
+async def health():
+    return {"status": "healthy"}
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)