Update app.py

app.py

@@ -32,6 +32,21 @@ custom_css = """
     background-color: rgba(30, 30, 30, 0.95);
     border: 1px solid rgba(255, 255, 255, 0.1);
 }
+/* Badge container styling */
+.badge-container {
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    gap: 15px;
+    padding: 15px;
+    background-color: rgba(255, 255, 255, 0.9);
+    backdrop-filter: blur(10px);
+    border-radius: 15px;
+    margin-bottom: 20px;
+}
+.dark .badge-container {
+    background-color: rgba(30, 30, 30, 0.9);
+}
 """
 
 # State variable to track current model
@@ -42,6 +57,19 @@ def switch_model(model_choice):
     return gr.update(visible=False), gr.update(visible=True), model_choice
 
 with gr.Blocks(fill_height=True, theme=gr.themes.Soft, css=custom_css) as demo:
+    # Add badges at the top
+    with gr.Row(elem_classes="badge-container"):
+        gr.HTML("""
+            <div style="display: flex; justify-content: center; align-items: center; gap: 15px;">
+                <a href="https://huggingface.co/spaces/VIDraft/gpt-oss-RAG" target="_blank">
+                    <img src="https://img.shields.io/static/v1?label=gpt-oss-20b&message=RAG&color=%23000080&labelColor=%23ffa500&logo=huggingface&logoColor=white&style=for-the-badge" alt="badge">
+                </a>
+                <a href="https://discord.gg/openfreeai" target="_blank">
+                    <img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=%23000080&labelColor=%23ffa500&logo=discord&logoColor=white&style=for-the-badge" alt="badge">
+                </a>
+            </div>
+        """)
+    
     with gr.Row():
         # Sidebar
         with gr.Column(scale=1):