Update main.py

main.py

@@ -1,8 +1,13 @@
 from fastapi import FastAPI
-from duckduckgo_search import DDGS
+
 from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
-
+import fitz  # PyMuPDF
+import requests
+import shutil
+import os
+from fastapi import FastAPI, File, UploadFile
+from uuid import uuid4
 app = FastAPI()
 origins = ["*"] 
 
@@ -13,25 +18,53 @@ app.add_middleware(
     allow_methods=["*"], 
     allow_headers=["*"], 
 )
+UPLOAD_DIR = "uploads"
+IMAGES_DIR = "images"
+TELEGRAPH_UPLOAD_URL = "https://telegra.ph/upload"
+
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+os.makedirs(IMAGES_DIR, exist_ok=True)
+
+@app.get("/")
+def check_status():
+    return {"status": "API is working"}
+
+@app.post("/upload")
+def upload_pdf(file: UploadFile = File(...)):
+    file_path = os.path.join(UPLOAD_DIR, f"{uuid4()}.pdf")
+    with open(file_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+    
+    image_links = convert_pdf_to_images(file_path)
+    return {"images": image_links}
+
+def convert_pdf_to_images(pdf_path):
+    doc = fitz.open(pdf_path)
+    image_links = []
+    
+    for i, page in enumerate(doc):
+        image_path = os.path.join(IMAGES_DIR, f"{uuid4()}.png")
+        pix = page.get_pixmap()
+        pix.save(image_path)
+        
+        telegraph_url = upload_to_telegraph(image_path)
+        if telegraph_url:
+            image_links.append(telegraph_url)
+        
+        os.remove(image_path)  # Cleanup after upload
+    
+    os.remove(pdf_path)  # Cleanup uploaded PDF
+    return image_links
+
+def upload_to_telegraph(image_path):
+    with open(image_path, "rb") as img_file:
+        response = requests.post(TELEGRAPH_UPLOAD_URL, files={"file": img_file})
+    
+    if response.status_code == 200 and isinstance(response.json(), list):
+        return "https://telegra.ph" + response.json()[0]["src"]
+    return None
+
 
-def chat_with_model(query: str, model: str) -> JSONResponse:
-    results = None
-    try:
-        results = DDGS().chat(query, model=model)
-    except Exception as e:
-        return JSONResponse(content={"error": str(e)}, status_code=500)
-    return JSONResponse(content={"results": results})
-
-@app.get("/chat/")
-async def chat(query: str) -> JSONResponse:
-    results = None
-    try:
-        return  DDGS().chat(query, model='gpt-4o-mini')
-    except Exception as e:
-        try:
-            return  DDGS().chat(query, model='claude-3-haiku')
-        except Exception as e:
-            return JSONResponse(content={"error": str(e)})
 
 if __name__ == "__main__":
     import uvicorn