Update app/fir_pdf_gen.py

app/fir_pdf_gen.py

@@ -5,16 +5,6 @@ from pydantic import BaseModel
 import os
 import uuid
 import httpx
-import cloudinary
-import cloudinary.uploader
-
-# Initialize Cloudinary
-cloudinary.config( 
-    cloud_name = "dgdxa7qqg",  # Your Cloudinary cloud name
-    api_key = "376418913322648",  # Your Cloudinary API key
-    api_secret = "ut-74eisi_NAFxfrEUDhER2szgM",  # Your Cloudinary API secret
-    secure=True
-)
 
 router = APIRouter()
 
@@ -29,7 +19,7 @@ def generate_fir_pdf(data: dict) -> str:
     pdf.add_page()
     pdf.set_font("Arial", size=11)
     
-    # Add content to the PDF
+    # Add content
     pdf.cell(0, 10, f"Book No.: {data['book_no']}", ln=True)
     pdf.cell(0, 10, f"Form No.: {data['form_no']}", ln=True)
     pdf.cell(0, 10, f"Police Station: {data['police_station']}", ln=True)
@@ -44,13 +34,12 @@ def generate_fir_pdf(data: dict) -> str:
     pdf.cell(0, 10, f"Date and Time of Dispatch from Police Station: {data['dispatch_time']}", ln=True)
     pdf.cell(0, 10, f"Signature of Writer: ..............................", ln=True)
     
-    # Save PDF to a temporary file
+    # Save PDF
     output_dir = "fir_reports"
     os.makedirs(output_dir, exist_ok=True)
     file_name = f"FIR_Report_{uuid.uuid4().hex}.pdf"
     file_path = os.path.join(output_dir, file_name)
     pdf.output(file_path)
-
     return file_path
 
 class FIRDetails(BaseModel):
@@ -69,7 +58,6 @@ class FIRDetails(BaseModel):
 
 @router.get("/download/{file_name}")
 async def download_file(file_name: str):
-    # Check if the file exists before returning
     file_path = os.path.join("fir_reports", file_name)
     if not os.path.exists(file_path):
         raise HTTPException(status_code=404, detail="File not found")
@@ -79,49 +67,33 @@ async def get_lawgpt_response(description_offense: str) -> str:
     """
     Sends the description_offense to an external service and retrieves the response.
     """
-    url = "http://0.0.0.0:7860/lawgpt/process-fir-description/"  # Adjust URL based on actual route
+    url = "http://0.0.0.0:7860/lawgpt/chat"  # Replace with the actual URL
     try:
         async with httpx.AsyncClient() as client:
-            response = await client.post(url, json={"description_offense": description_offense})
-            response.raise_for_status()  # Raise error for status codes >= 400
+            response = await client.post(url, json={"question": description_offense})
+            response.raise_for_status()  # Raise an error for HTTP codes >= 400
             data = response.json()
-            return data.get("processed_description", description_offense)  # Return processed description
+            return data.get("response", description_offense)  # Use original if no response
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to get response from LawGPT: {str(e)}")
 
 @router.post("/")
 async def generate_fir(details: FIRDetails):
     try:
-        # Get the processed description from LawGPT
+        # Get response from LawGPT for description_offense
         updated_description = await get_lawgpt_response(details.description_offense)
+        
+        # Replace the description_offense with the processed response
         details.description_offense = updated_description
 
-        # Generate the FIR PDF
+        # Generate PDF with the updated description
         file_path = generate_fir_pdf(details.dict())
-
-        # Upload the PDF to Cloudinary
-        try:
-            response = cloudinary.uploader.upload(
-                file_path,
-                resource_type="raw",  # PDFs are treated as raw files in Cloudinary
-                folder="fir_reports/"
-            )
-            os.remove(file_path)  # Clean up the local file
-
-            # Get Cloudinary URLs
-            view_url = response['secure_url']
-            download_url = f"{view_url}?attachment=true"
-
-            return {
-                "message": "FIR PDF generated and uploaded successfully!",
-                "view_url": view_url,  # Cloudinary view URL
-                "download_url": download_url  # Cloudinary download URL
-            }
-        except Exception as e:
-            os.remove(file_path)  # Clean up in case of failure
-            raise HTTPException(status_code=500, detail=f"Error uploading to Cloudinary: {str(e)}")
-
+        
+        return {
+            "message": "FIR PDF generated successfully!",
+            "download_url": f"http://0.0.0.0:7860/generate-fir/download/{os.path.basename(file_path)}"
+        }
     except HTTPException as http_exc:
         raise http_exc
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))