import gradio as gr
from flask import Flask, request
from transformers import RobertaForSequenceClassification, RobertaTokenizer
import torch
import os
app = Flask(__name__)

ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
tokenizer = RobertaTokenizer.from_pretrained("PirateXX/Hindi-AI-Detector", use_auth_token= ACCESS_TOKEN)
model = RobertaForSequenceClassification.from_pretrained("PirateXX/Hindi-AI-Detector", use_auth_token= ACCESS_TOKEN)

from torch import cuda
device = 'cuda' if cuda.is_available() else 'cpu'

def predict(query):
    query = " ".join(query.split())
    tokens = tokenizer.encode(query)
    all_tokens = len(tokens)
    tokens = tokens[:tokenizer.model_max_length - 2]
    used_tokens = len(tokens)
    tokens = torch.tensor([tokenizer.bos_token_id] + tokens + [tokenizer.eos_token_id]).unsqueeze(0)
    mask = torch.ones_like(tokens)

    with torch.no_grad():
        logits = model(tokens.to(device), attention_mask=mask.to(device))[0]
        probs = logits.softmax(dim=-1)

    fake, real = probs.detach().cpu().flatten().numpy().tolist()
    return {"Real": real, "Fake":fake}

demo = gr.Interface(
        fn=predict, 
        inputs=gr.Textbox(placeholder="Copy and paste here..."), 
        article = "Visit <a href = \"https://ai-content-detector.online/\">AI Content Detector</a> for analysis on english text!",
        outputs="label",
        # interpretation="default",
        examples=["महेंद्र सिंह धोनी, जिसे हम सभी प्यार से \"धोनी\" कहते हैं, भारतीय क्रिकेट को अपनी अद्वितीय पहचान बनाने वाले एक अद्वितीय खिलाड़ी हैं। उन्होंने अपनी प्रशंसा को वर्ल्ड क्रिकेट में कार्यकारी रूप से सिद्ध किया है। उनकी कैप्टनी में भारतीय टीम ने 2007 टी-20 विश्व कप, 2010 एशिया कप और 2011 वनडे विश्व कप जीते हैं। धोनी को उनकी ठोस कैप्टनी, अद्वितीय बैटिंग और शानदार कीपिंग कौशल के लिए जाना जाता है। उनकी शांति, स्थिरता और खुशमिजाजी का भी महत्वपूर्ण योगदान है, जो उन्हें दिग्गज कप्तान के रूप में अद्वितीय बनाता है। धोनी अब संन्यास ले चुके हैं, लेकिन उनका प्रभाव और महत्व अब भी क्रिकेट जगत में दिखाई देता है। वह हमेशा एक युगपुरुष और देशभक्त की भूमिका में चमक रहे हैं और हमेशा भारतीय क्रिकेट के सबसे प्रिय और सम्मानित व्यक्तित्वों में गिने जाएंगे।", "भाई-भाई के बीच होने वाले प्यार और नफरत को भारतीयों ने काफी पसंद किया है। महाभारत से लेकर तो अमर अकबर एंथोनी, दीवार, करण-अर्जुन तक दर्शकों को यह भाइयों की कहानी बेहद पसंद आई है। इसी बात को ध्यान में रखते हुए दोनों करणों (निर्माता करण जौहर और निर्देशक करण मल्होत्रा) को हॉलीवुड फिल्म 'वॉरियर' में एक सफल रीमेक बनाने के गुण नजर आए। यह भी दो भाइयों की कहानी है जिसमें भरपूर इमोशन और एक्शन है, लेकिन इसका बॉलीवुड संस्करण करने में 'ब्रदर्स' के मेकर्स मात खा गए। बदले में हमें एक लंबी, घिसटती हुई और उबाऊ फिल्म देखने को मिलती है।"])

demo.launch(show_api=False)