Add application file

app.py

@@ -0,0 +1,65 @@
+import gradio as gr
+import json
+from mistralai.client import MistralClient
+from mistralai.models.chat_completion import ChatMessage
+import os
+
+client = MistralClient(api_key=os.environ["MISTRAL_API_KEY"])
+model = "open-mixtral-8x7b"
+
+messages = {
+    'English': {
+        'system_message': "You are a moderation assistant. Given a {}, determine if it should be flagged as spam, not spam, or possibly spam. Respond with a JSON object containing a 'spam' field with a string value of 'true', 'false', or 'possible'.",
+        'user_message': "{}: {}\\n\\nShould this {} be flagged as spam? Respond with a JSON object."
+    },
+    'Spanish': {
+        'system_message': "Eres un asistente de moderación. Dado un {}, determina si debe marcarse como spam, no spam o posiblemente spam. Responde con un objeto JSON que contenga un campo 'spam' con un valor de cadena de 'true', 'false' o 'possible'.",
+        'user_message': "{}: {}\\n\\n¿Se debe marcar este {} como spam? Responde con un objeto JSON."
+    },
+    'French': {
+        'system_message': "Vous êtes un assistant de modération. Étant donné un {}, déterminez s'il doit être signalé comme spam, non spam ou éventuellement spam. Répondez avec un objet JSON contenant un champ 'spam' avec une valeur de chaîne 'true', 'false' ou 'possible'.",
+        'user_message': "{}: {}\\n\\nCe {} doit-il être signalé comme spam ? Répondez avec un objet JSON."
+    }
+}
+
+def make_request(client, model, system_message, user_message, temperature=0.7, top_p=0.6, max_tokens=50):
+    messages = [
+        ChatMessage(role="system", content=system_message),
+        ChatMessage(role="user", content=user_message)
+    ]
+    response = client.chat(model=model, messages=messages, temperature=temperature, top_p=top_p, max_tokens=max_tokens)
+    return response.choices[0].message.content if response.choices else ""
+
+def moderate_content(content, content_type, moderation_rules, language):
+    system_message = messages[language]['system_message'].format(content_type.lower())
+    user_message = messages[language]['user_message'].format(content_type, content, content_type.lower())
+    
+    response = make_request(client, model, system_message, user_message, temperature=0.3, top_p=0.6, max_tokens=50)
+    try:
+        response_json = json.loads(response)
+        if 'spam' in response_json:
+            spam_status = response_json['spam'].lower()
+            if spam_status == 'true':
+                return "Spam"
+            elif spam_status == 'false':
+                return "Not Spam"
+            elif spam_status == 'possible':
+                return "Possibly Spam"
+    except (json.JSONDecodeError, KeyError, ValueError):
+        pass
+    return "Unable to determine spam status"
+
+content_type = gr.Radio(["Proposal", "Comment"], label="Content Type")
+content = gr.Textbox(label="Content")
+moderation_rules = gr.Textbox(label="Moderation Rules")
+language = gr.Dropdown(["English", "Spanish", "French"], label="Language")
+
+interface = gr.Interface(
+    fn=moderate_content,
+    inputs=[content, content_type, moderation_rules, language],
+    outputs="text",
+    title="Content Moderation",
+    description="Enter a proposal or comment, along with moderation rules, to determine if it's spam.",
+)
+
+interface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+mistralai
+os