Update app.py

.env.example

@@ -0,0 +1,15 @@
+# Whether to use the HuggingFace API or Ollama
+# Set this to yes or no
+USE_HUGGINGFACE=
+
+# This token is optional but you'll have more restrictive rate limits and model options without it
+# Get your HuggingFace token from https://huggingface.co/settings/tokens
+HUGGINGFACE_API_TOKEN=
+
+# The model ID to use from HuggingFace for the tool calling LLM
+# Example: meta-llama/Llama-3.3-70B-Instruct
+TOOL_MODEL_ID=
+
+AGENT_ENDPOINT=
+
+AGENT_KEY=

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+.env

README.md

@@ -0,0 +1,14 @@
+---
+title: Vie Publique
+emoji: 💬
+colorFrom: yellow
+colorTo: purple
+sdk: gradio
+sdk_version: 5.0.1
+app_file: app.py
+pinned: false
+license: mit
+short_description: Chat with vie-publique.sn
+---
+
+An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).

app.py

@@ -0,0 +1,105 @@
+from dotenv import load_dotenv
+import gradio as gr
+import openai
+import os
+
+load_dotenv()
+
+MAX_HISTORY_LENGTH = 10
+
+def load_css():
+    try:
+        with open('custom.css', 'r', encoding='utf-8') as file:
+            return file.read()
+    except FileNotFoundError:
+        logging.warning("⚠️ custom.css non trouvé, chargement ignoré.")
+        return ""
+
+def format_history_for_api(history):
+    """
+    Convertit l'historique Gradio en format OpenAI.
+    """
+    formatted_messages = []
+    for message in history:
+        formatted_messages.extend([
+            {"role": "user", "content": str(message[0])},
+            {"role": "assistant", "content": str(message[1])}
+        ])
+    return formatted_messages
+
+def rag_with_reasoner(user_query: str, history: list) -> dict:
+    """
+    Outil RAG qui prend une requête utilisateur et renvoie une réponse basée sur le contexte historique.
+    """
+    client = openai.OpenAI(
+        base_url=os.getenv("AGENT_ENDPOINT"),
+        api_key=os.getenv("AGENT_KEY"),
+    )
+
+    # Formatage des messages pour l'API
+    messages = format_history_for_api(history) if history else []
+    messages.append({"role": "user", "content": str(user_query)})
+
+    try:
+        response = client.chat.completions.create(
+            model="n/a",
+            messages=messages
+        )
+        return {"role": "assistant", "content": response.choices[0].message.content}
+    except Exception as e:
+        print(f"Error calling OpenAI API: {e}")
+        return {"role": "assistant", "content": f"Une erreur s'est produite: {str(e)}"}
+
+def chat_function(message, history):
+    """
+    Fonction de chat pour Gradio.
+    """
+    try:
+        # Limiter la taille de l'historique
+        if len(history) > MAX_HISTORY_LENGTH:
+            history = history[-MAX_HISTORY_LENGTH:]
+
+        response = rag_with_reasoner(message, history)
+        return response["content"]
+    except Exception as e:
+        print(f"Error in chat function: {e}")
+        return f"Une erreur s'est produite: {str(e)}"
+
+# Configuration de l'interface Gradio
+css = """
+.message.tool-call { background-color: #f0f8ff; }
+.message.tool-result { background-color: #f5f5f5; }
+.message.error { background-color: #fff0f0; }
+.chat-header {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    gap: 10px;
+    background: linear-gradient(90deg, #0000d3, #010272);
+}
+.chat-header img {
+    height: 40px;
+}
+"""
+
+# Création de l'interface de chat avec Gradio
+chat_interface = gr.ChatInterface(
+    fn=chat_function,
+    title="<div class='chat-header'><img src='https://www.vie-publique.sn/_nuxt/vie-publique-logo-4.BGl8Bhgm.svg' alt='Logo'><span>🤖</span></div>",
+    description="Discutez avec un agent IA pour des analyses précises et des conseils sur la vie publique du Sénégal 🇸🇳",
+    examples=["Sénégal 2050", "Budget 2025", "Constitution du Sénégal", "Rapports publics"],
+    textbox=gr.Textbox(
+        placeholder="Posez votre question ici...",
+        submit_btn="Envoyer 📤",
+        container=False,
+        scale=7
+    ),
+    theme=gr.themes.Soft(),
+    css=css + load_css()
+)
+
+def main():
+    chat_interface.launch()
+
+if __name__ == "__main__":
+    main()

custom.css

@@ -0,0 +1,119 @@
+
+/* Custom theme colors */
+:root {
+    --primary-color: #0000d3;
+    --secondary-color: #c3c3f5;
+    --background-color: #f8fafc;
+    --text-color: white;
+    --text-secondary: #000050;
+  }
+  
+  @media (prefers-color-scheme: dark) {
+    :root {
+      --primary-color: #c3c3f5;
+      --secondary-color: #0000d3;
+      --background-color: #1e1e2e;
+      --text-color: white;
+      --text-secondary: #000050;
+    }
+  
+    .chat-interface {
+      background: rgba(30, 30, 46, 0.8) !important;
+    }
+  
+    .message {
+      background: rgba(40, 40, 56, 0.8) !important;
+      color: var(--text-color) !important;
+    }
+  
+    .input-box {
+      background: rgba(30, 30, 46, 0.9) !important;
+      color: var(--text-color) !important;
+    }
+  }
+  
+  /* Main container styling */
+  .gradio-container {
+    /* background-color: var(--background-color) !important */
+    background-image: url('https://gradientsss.app/mesh.svg');
+    background-attachment: fixed;
+    background-size: cover;
+  }
+  
+  /* Chat interface styling */
+  .chat-interface {
+    max-width: 800px;
+    margin: 0 auto;
+    padding: 2rem;
+    border-radius: 1rem;
+    background: rgba(255, 255, 255, 0.8);
+    backdrop-filter: blur(10px);
+    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+  }
+  
+  @media (prefers-color-scheme: dark) {
+    .chat-interface {
+      background: rgba(30, 30, 46, 0.8) !important;
+    }
+    
+    .gradio-container {
+      background-color: #1e1e2e !important;
+    }
+    
+    .message-wrap {
+      background: rgba(40, 40, 56, 0.8) !important;
+    }
+  }
+  
+  /* Title styling */
+  .title {
+    color: var(--primary-color);
+    font-size: 2.5rem;
+    font-weight: bold;
+    text-align: center;
+    margin-bottom: 1rem;
+  }
+  
+  .title span {
+    font-size: 24px;
+    font-weight: 600;
+  }
+  
+  /* Message container */
+  .message {
+    padding: 1rem;
+    margin: 0.5rem 0;
+    border-radius: 0.5rem;
+    color: var(--text-color) !important;
+  }
+  
+  /* Input box styling */
+  .input-box {
+    border: 2px solid var(--primary-color);
+    border-radius: 0.5rem;
+    padding: 0.75rem;
+    background: white;
+  }
+  
+  /* Button styling */
+  button {
+    background: var(--primary-color) !important;
+    color: var(--text-color) !important;
+    border-radius: 0.5rem !important;
+    padding: 0.75rem 1.5rem !important;
+    transition: all 0.3s ease !important;
+  }
+
+  button p {
+    color: var(--text-color) !important;
+  }
+  
+  button:hover {
+    background: var(--secondary-color) !important;
+    transform: translateY(-2px);
+  }
+  
+  button:hover p {
+    color: var(--text-secondary) !important;
+  }
+  /* footer {visibility: hidden} */

requirements.txt

@@ -0,0 +1,3 @@
+smolagents
+openai
+gradio