app.py

from flask import Flask, render_template, request, jsonify
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

app = Flask(__name__)

# Load your specific SciBERT model from HF
MODEL_PATH = "gsstec/aegis-scibert-technical"
tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/predict', methods=['POST'])
def predict():
    data = request.json
    year = data.get("year", "2026")
    
    # Construct technical context for SciBERT for different tech categories
    # Comprehensive categories that can disrupt economy, lead to war, or cause diseases
    categories = [
        # AI & Machine Learning (Economic Disruption & Warfare)
        "Artificial Intelligence", "Machine Learning", "Deep Learning", "Neural Networks",
        "Computer Vision", "Natural Language Processing", "Reinforcement Learning", "AGI",
        "AI Automation", "Autonomous Systems", "AI Surveillance", "Predictive Analytics",
        
        # Robotics & Automation (Economic & Military)
        "Industrial Robotics", "Military Robotics", "Autonomous Weapons", "Drone Technology",
        "Swarm Robotics", "Humanoid Robots", "Surgical Robots", "Combat Robots",
        
        # Quantum Technologies (Security & Computing)
        "Quantum Computing", "Quantum Cryptography", "Quantum Communications", "Quantum Sensing",
        "Quantum Radar", "Quantum Internet", "Post-Quantum Cryptography", "Quantum Supremacy",
        
        # Biotechnology & Genetic Engineering (Disease & Bioweapons)
        "Gene Editing", "CRISPR", "Synthetic Biology", "Bioweapons", "Gain-of-Function Research",
        "Viral Engineering", "Bacterial Weapons", "Toxin Production", "Genetic Warfare",
        "Bioterrorism", "Pathogen Enhancement", "Designer Pathogens", "Dual-Use Research",
        
        # Cyber Warfare & Information Security
        "Cyber Warfare", "Information Warfare", "Cyberterrorism", "Advanced Persistent Threats",
        "Zero-Day Exploits", "Ransomware", "State-Sponsored Hacking", "Critical Infrastructure Attacks",
        "Election Interference", "Deepfakes", "Social Media Manipulation", "Propaganda AI",
        
        # Nuclear & WMD Technologies
        "Nuclear Weapons", "Nuclear Proliferation", "Enrichment Technology", "Plutonium Production",
        "Miniaturized Warheads", "Tactical Nuclear Weapons", "Dirty Bombs", "Nuclear Terrorism",
        "Fusion Weapons", "Nuclear Delivery Systems", "ICBM Technology", "Nuclear Submarines",
        
        # Chemical & Biological Weapons
        "Chemical Weapons", "Nerve Agents", "Biological Weapons", "Toxin Weapons",
        "Aerosol Delivery", "Binary Weapons", "Novichok", "VX Gas", "Anthrax Weapons",
        "Smallpox Weapons", "Hemorrhagic Fever Weapons", "Weaponized Viruses",
        
        # Space & Satellite Technologies (Military)
        "Anti-Satellite Weapons", "Space Warfare", "Satellite Jamming", "Space Debris Weapons",
        "Kinetic Bombardment", "Space-Based Weapons", "GPS Disruption", "Communication Satellites",
        "Reconnaissance Satellites", "Military Space Stations", "Lunar Military Bases",
        
        # Hypersonic & Advanced Weapons
        "Hypersonic Missiles", "Hypersonic Glide Vehicles", "Scramjet Technology", "Mach 5+ Weapons",
        "Boost-Glide Systems", "Hypersonic Cruise Missiles", "Prompt Global Strike",
        
        # Directed Energy & Electromagnetic Weapons
        "Laser Weapons", "Microwave Weapons", "EMP Weapons", "Electromagnetic Pulse",
        "High-Power Microwave", "Particle Beam Weapons", "Plasma Weapons", "Rail Guns",
        "Coil Guns", "Electromagnetic Launchers", "Electronic Warfare Systems",
        
        # Nanotechnology & Advanced Materials
        "Military Nanotechnology", "Nanoweapons", "Smart Materials", "Metamaterials",
        "Self-Assembling Materials", "Programmable Matter", "Molecular Manufacturing",
        "Carbon Nanotubes", "Graphene Applications", "Stealth Materials", "Armor Technology",
        
        # Communication & Network Technologies
        "5G/6G Networks", "Quantum Internet", "Mesh Networks", "Satellite Internet",
        "Underwater Communications", "Military Communications", "Encrypted Communications",
        "Network Warfare", "Internet Infrastructure", "Fiber Optic Networks",
        
        # Surveillance & Intelligence Technologies
        "Mass Surveillance", "Facial Recognition", "Biometric Tracking", "Location Tracking",
        "Behavioral Analytics", "Predictive Policing", "Social Credit Systems", "Digital Twins",
        "IoT Surveillance", "Smart City Surveillance", "Drone Surveillance", "Satellite Surveillance",
        
        # Economic & Financial Disruption Technologies
        "Algorithmic Trading", "High-Frequency Trading", "Market Manipulation AI", "Flash Crashes",
        "Cryptocurrency Warfare", "Digital Currency Control", "Economic Sanctions Technology",
        "Supply Chain Disruption", "Critical Resource Control", "Energy Weapon Economics",
        
        # Climate & Environmental Weapons
        "Weather Modification", "Geoengineering", "Climate Weapons", "Drought Induction",
        "Hurricane Modification", "Earthquake Weapons", "Tsunami Generation", "Volcanic Triggering",
        "Ozone Depletion", "Agricultural Disruption", "Water Poisoning", "Soil Contamination",
        
        # Medical & Pharmaceutical Warfare
        "Targeted Bioweapons", "Ethnic Bioweapons", "Pharmaceutical Warfare", "Drug Resistance",
        "Vaccine Disruption", "Medical Supply Disruption", "Hospital Cyberattacks", "Medical Device Hacking",
        "Pandemic Engineering", "Disease Surveillance Evasion", "Antibiotic Resistance Engineering",
        
        # Energy & Infrastructure Disruption
        "Power Grid Attacks", "Energy Infrastructure", "Smart Grid Vulnerabilities", "Pipeline Attacks",
        "Nuclear Plant Attacks", "Renewable Energy Disruption", "Battery Technology Warfare",
        "Critical Mineral Control", "Rare Earth Monopolization", "Energy Storage Attacks",
        
        # Transportation & Logistics Disruption
        "Autonomous Vehicle Hacking", "Traffic System Attacks", "Port Disruption", "Aviation Attacks",
        "Maritime Warfare", "Supply Chain Attacks", "Shipping Disruption", "Railroad Attacks",
        "GPS Spoofing", "Navigation Disruption", "Logistics AI Warfare",
        
        # Food & Agriculture Warfare
        "Agricultural Bioweapons", "Crop Diseases", "Livestock Diseases", "Pollinator Destruction",
        "Soil Degradation", "Water Table Poisoning", "Seed Monopolization", "Fertilizer Disruption",
        "Food Supply Chain Attacks", "Precision Agriculture Hacking", "Vertical Farming Disruption",
        
        # Psychological & Cognitive Warfare
        "Psychological Operations", "Cognitive Warfare", "Mind Control Technology", "Subliminal Messaging",
        "Neurotechnology Weapons", "Brain-Computer Interface Hacking", "Memory Manipulation",
        "Behavior Modification", "Mass Hypnosis", "Neurological Weapons", "Psychoactive Weapons",
        
        # Emerging & Exotic Technologies
        "Metamaterial Cloaking", "Invisibility Technology", "Teleportation Research", "Antigravity Research",
        "Exotic Matter", "Dark Energy Weapons", "Dimensional Weapons", "Time Manipulation",
        "Consciousness Transfer", "Digital Immortality", "Synthetic Life", "Artificial Evolution"
    ]
    tech_scores = {}
    
    for category in categories:
        # Create category-specific input text
        input_text = f"Scientific and technological advancements in {category} emergent in the year {year}."
        
        # Tokenization
        inputs = tokenizer(input_text, return_tensors="pt", truncation=True, max_length=512)
        
        # Prediction
        with torch.no_grad():
            outputs = model(**inputs)
            # Get the first prediction score for this category
            prediction = torch.softmax(outputs.logits, dim=1).tolist()[0]
            tech_scores[category] = prediction[0]

    return jsonify({
        "year": year,
        "tech_scores": tech_scores,
        "status": "SENT_TO_CONDUCTOR"
    })

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)