Create main.py

main.py

@@ -0,0 +1,43 @@
+from fastapi import FastAPI, HTTPException
+from rag_service import rag_service
+
+app = FastAPI(
+    title="Multi-Modal RAG API",
+    description="API for querying documents with text and images",
+    version="1.0.0"
+)
+
+@app.get("/")
+async def root():
+    return {"message": "Multi-Modal RAG API is running"}
+
+@app.post("/ask")
+async def ask_question(question: str):
+    """
+    Ask a question to the RAG system
+    """
+    try:
+        response = rag_service.ask_question(question)
+        return {
+            "question": question,
+            "response": response,
+            "status": "success"
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error processing question: {str(e)}")
+
+@app.get("/health")
+async def health_check():
+    """
+    Health check endpoint
+    """
+    try:
+        # Check if vectorstore is accessible
+        count = rag_service.vectorstore._collection.count()
+        return {
+            "status": "healthy",
+            "vectorstore_documents": count,
+            "docstore_documents": len(rag_service.store.store)
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Health check failed: {str(e)}")