Upload 2 files

gradio_iface.py

@@ -0,0 +1,91 @@
+import requests
+import os
+import json
+import gradio as gr
+from dotenv import load_dotenv
+
+load_dotenv()
+
+api_key = os.getenv('API_KEY')
+
+
+# Mapping function to class
+function_mapping = {
+    "Creation_dossier_kbis": 'POST',
+    "Redaction_non_juridique": 'DRAFT',
+    "Redaction_juridique": '0',
+    "Resume": '0',
+    "Salutations": 'DRAFT',
+    "Traduction": 'DRAFT',
+    "Information_utilisateur": 'GET',
+    "Information_dossier": 'GET',
+    "Information_personne_societe": 'GET',
+    "Autre_demande": '0'
+}
+
+# Load tools
+with open("tools-intent-detection.json", "r", encoding="utf-8") as file:
+    tools = json.load(file)
+
+# API configuration
+url = 'https://openai-dev-fra-001.openai.azure.com/openai/deployments/gpt-4o/chat/completions?api-version=2024-10-21'
+headers = {
+    'api-key': api_key,
+    'Content-Type': 'application/json'
+}
+
+def get_model_response(user_prompt):
+    # messages list
+    messages = [
+        {
+            "role": "system",
+            "content": ""
+        },
+        {
+            "role": "user",
+            "content": user_prompt
+        }
+    ]
+    
+    # payload
+    data = {
+        "model": "gpt-4o",
+        "messages": messages,
+        "tools": tools,
+        "tool_choice": "required"
+    }
+    
+    # API call
+    response = requests.post(url, headers=headers, data=json.dumps(data))
+    
+    # Process the response
+    if response.status_code != 200:
+        return f"Error: {response.status_code} - {response.text}", "0"
+    else:
+        response_data = response.json()
+        if 'choices' in response_data:
+            reply = response_data["choices"][0]["message"]
+            
+            # Extract function name from tool_calls
+            if 'tool_calls' in reply and len(reply['tool_calls']) > 0:
+                function_name = reply['tool_calls'][0]['function']['name']
+                # Get the corresponding class from the mapping
+                category = function_mapping.get(function_name, "0")  # 0 is default if function name not found
+                return function_name, category
+            else:
+                return "No function called", "0"
+        else:
+            return "Unexpected response format.", "0"
+
+# Gradio app
+iface = gr.Interface(
+    title="Intent Detection Playground",
+    fn=get_model_response,
+    inputs=gr.Textbox(label="User Prompt"),
+    outputs=[
+        gr.Textbox(label="Intention"),
+        gr.Textbox(label="Category")
+    ]
+)
+
+iface.launch()

tools-intent-detection.json

@@ -0,0 +1,82 @@
+[
+    {
+      "type": "function",
+      "function": {
+        "name": "Creation_dossier_kbis",
+        "description": "L'utilisateur partage un extrait de kbis ou souhaite créer un dossier",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    },
+    {
+      "type": "function",
+      "function": {
+        "name": "Redaction_non_juridique",
+        "description": "Redaction de correspondances exclusivement non juridiques (email, courrier, lettre, newsletter ou publication réseaux sociaux)",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    },
+    {
+      "type": "function",
+      "function": {
+        "name": "Redaction_juridique",
+        "description": "Redaction de contenu juridique",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    },
+    {
+      "type": "function",
+      "function": {
+        "name": "Resume",
+        "description": "Resume un texte",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    },
+    {
+      "type": "function",
+      "function": {
+        "name": "Salutations",
+        "description": "Reponse aux salutations et demande les fonctionnalités de l'assistant IA",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    },
+    {
+      "type": "function",
+      "function": {
+        "name": "Traduction",
+        "description": "Traduction de contenu en une autre langue",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    },
+    {
+      "type": "function",
+      "function": {
+        "name": "Information_utilisateur",
+        "description": "Informations sur les utilisateurs (id, type, etc.)",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    }, 
+    {
+      "type": "function",
+      "function": {
+        "name": "Information_dossier",
+        "description": "Informations generales et financieres sur les dossiers",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    },
+    {
+      "type": "function",
+      "function": {
+        "name": "Information_personne_societe",
+        "description": "Informations et factures d'une personne ou une société",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    },
+    {
+      "type": "function",
+      "function": {
+        "name": "Autre_demande",
+        "description": "Traite toutes les demandes qui ne sont pas couvertes par les autres fonctions",
+        "parameters": { "type": "object", "properties": {} }
+      }
+    }
+  ]