Create app.py

app.py

@@ -0,0 +1,131 @@
+from unittest import result
+from pdfplumber import pdf
+import streamlit as st
+import requests
+from crypto import encrypt_pdf, decrypt_pdf
+import os
+import time
+import json, asyncio
+from transformers import pipeline
+
+api_keys = {}
+with open('api_keys.json', 'r') as file:
+    api_keys = json.load(file)
+
+PINATA_API_KEY = api_keys['PINATA_API_KEY']
+PINATA_SECRET_API_KEY = api_keys['PINATA_SECRET_API_KEY']
+PINATA_API_URL = "https://api.pinata.cloud/pinning/pinFileToIPFS"
+IPFS_GATEWAY = "https://gateway.pinata.cloud/ipfs/"
+
+with open("university_keys.json", "r") as f:
+    UNIVERSITY_KEYS = json.load(f)
+
+@st.cache_resource  
+def load_ai_checker():  
+    return  pipeline("text-generation", model="distilgpt2")
+
+ai_checker = load_ai_checker()
+
+def run_ai_check(pdf_text):
+    prompt = "Does this certificate look fake? Answer with 'Yes' or 'No': " + pdf_text[:500]
+    result = ai_checker(prompt, max_length=512, do_sample=False)[0]['generated_text']
+    print(result, "AI Check \n")
+    return "No" in result
+
+def upload_to_ipfs(file_bytes, filename):
+    headers = {
+        "pinata_api_key": PINATA_API_KEY,
+        "pinata_secret_api_key": PINATA_SECRET_API_KEY
+    }
+    files = {"file": (filename, file_bytes, "application/pdf")}
+    response = requests.post(PINATA_API_URL, headers=headers, files=files)
+    if response.status_code == 200:
+        return response.json().get("IpfsHash")
+    return None
+
+if __name__ == "__main__":
+    st.title("🎓 Certificate Verification System")
+    tabs = ["Sign Certificate", "Encrypt Certificate", "Verify Certificate", "Decrypt Certificate"]
+    choice = st.sidebar.radio("Select Action", tabs)
+
+    if choice == "Sign Certificate":
+        st.header("Sign & Encrypt Certificate")
+        university = st.selectbox("Select University", list(UNIVERSITY_KEYS.keys()))
+        pdf_file = st.file_uploader("Upload Certificate PDF", type=["pdf"])
+        sign_button = st.button("Sign and Encrypt Certificate")
+        
+        if pdf_file and university and sign_button:
+            pdf_bytes = pdf_file.read()
+            # AI Check
+            ai_check_passed = run_ai_check(pdf_bytes.decode(errors='ignore'))
+            if not ai_check_passed:
+                st.error("❌ AI Check Failed: Fake or Invalid Certificate Detected!")
+            else:
+                encrypted_pdf = encrypt_pdf(pdf_bytes, UNIVERSITY_KEYS[university].encode())
+                filename = f"{university.replace(' ', '_')}_{time.time()}_signed_and_encrypted.pdf"
+                ipfs_hash = upload_to_ipfs(encrypted_pdf, filename)
+                if ipfs_hash:
+                    st.success(f"✅ Certificate Signed & Stored on IPFS! URL: {IPFS_GATEWAY}{ipfs_hash}")
+                    st.download_button("Download Signed Certificate", encrypted_pdf, filename)
+                else:
+                    st.error("Failed to upload encrypted certificate to IPFS.")
+    elif choice == "Encrypt Certificate":
+        st.header("Encrypt Certificate")
+        university = st.selectbox("Select University", list(UNIVERSITY_KEYS.keys()))
+        pdf_file = st.file_uploader("Upload Certificate PDF", type=["pdf"])
+        encrypt_button = st.button("Encrypt Certificate")
+
+        if pdf_file and university and encrypt_button:
+            pdf_bytes = pdf_file.read()
+            encrypted_pdf = encrypt_pdf(pdf_bytes, UNIVERSITY_KEYS[university].encode())
+            filename = f"{university.replace(' ', '_')}_encrypted.pdf"
+            st.success("✅ Certificate Encrypted Successfully!")
+            st.download_button("Download Encrypted Certificate", encrypted_pdf, filename)
+
+    elif choice == "Verify Certificate":
+        st.header("Verify Certificate")
+        university = st.selectbox("Select University", list(UNIVERSITY_KEYS.keys()))
+        ipfs_hash = st.text_input("Enter IPFS Hash of the Signed Certificate")
+        user_pdf_file = st.file_uploader("Upload Signed and Encrypted Certificate PDF", type=["pdf"])
+        _verify_button = st.button("Verify Certificate")
+
+        if ipfs_hash and university and _verify_button and user_pdf_file:
+            response = requests.get(f"{IPFS_GATEWAY}{ipfs_hash}")
+            if response.status_code == 200:
+                encrypted_pdf = response.content
+                try:
+                    user_pdf_file = user_pdf_file.read()
+                    print("I am working")
+                    if encrypted_pdf != user_pdf_file:
+                        st.error("❌ Verification Failed! Encrypted PDF does not match the IPFS Hash.")
+                        st.stop()
+                    decrypted_pdf = decrypt_pdf(encrypted_pdf, UNIVERSITY_KEYS[university].encode())
+                    st.success("Certificate Verified Successfully and Valid ✅")
+                    st.download_button("Download Decrypted Certificate", decrypted_pdf, "decrypted_certificate.pdf")
+                except:
+                    st.error("❌ Verification Failed! Wrong University Selected.")
+            else:
+                st.error("Invalid IPFS Hash or file not found.")
+    elif choice == "Decrypt Certificate":
+        st.header("Decrypt Certificate")
+        university = st.selectbox("Select University", list(UNIVERSITY_KEYS.keys()))
+        pdf_file = st.file_uploader("Upload Encrypted Certificate PDF", type=["pdf"])
+        decrypt_button = st.button("Decrypt Certificate")
+
+        if pdf_file and university and decrypt_button:
+            pdf_bytes = pdf_file.read()
+            try:
+                decrypted_pdf = decrypt_pdf(pdf_bytes, UNIVERSITY_KEYS[university].encode())
+                st.success("✅ Certificate Decrypted Successfully!")
+                st.download_button("Download Decrypted Certificate", decrypted_pdf, "decrypted_certificate.pdf")
+            except:
+                st.error("❌ Decryption Failed! Wrong University Selected.")
+
+    # Footer
+    st.markdown("---")
+    st.markdown("📌 **Built by:**")
+    st.text(f"Led By: Dr. M. Asha Kiran")
+    st.text(f"Developer: Gajja Aarthi Goud")
+    st.text(f"Roll Number: 23MC201A34")
+    st.text(f"Department: MCA, IT Department, Anurag University")
+    # st.run()