Update app.py

app.py

@@ -3,16 +3,20 @@ from typing import List, Dict
 import gradio as gr
 from dotenv import load_dotenv
 from openai import OpenAI
+import base64
+from embedder import EmbeddingModel
+from Reranker import Reranker
 
-# ---- Windows event loop fix ----
 if sys.platform.startswith("win"):
     try:
         asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
     except Exception:
         pass
 
-# ---- Env ----
+# Env 
 load_dotenv()
+APP_Name    = os.getenv("APP_Name", "Rag chatbot in Ghaymah documentation system")
+APP_Version = os.getenv("APP_Version", "1.0.0")
 API_KEY = os.getenv("API_KEY")
 HOST = os.getenv("HOST")
 Embed_Model_Name = os.getenv("EMBEDDING_MODEL_NAME")
@@ -24,14 +28,21 @@ RPM_LIMIT = 2
 MIN_SECONDS_BETWEEN = 30
 N_DIM = 384
 
-# ---- OpenAI client ----
+#  OpenAI client 
 client = None
 if API_KEY:
     client = OpenAI(api_key=API_KEY, base_url="https://genai.ghaymah.systems")
 
-# ---- Your RAG bits ----
-from embedder import EmbeddingModel
-from Reranker import Reranker
+CSS = """
+.app-header{display:flex;align-items:center;gap:12px;justify-content:center;margin:6px 0 16px}
+.app-header img{height:60px;border-radius:12px}
+.app-title{font-weight:800;font-size:28px;line-height:1.1}
+.app-sub{opacity:.7;font-size:14px}
+"""
+
+COMPANY_LOGO = "download.jpeg"   
+OWNER_NAME   = "ENG. Ahmed Yasser El Sharkawy"
+
 
 def safe_chat_complete(model: str, messages: List[Dict], max_tokens: int = 9000) -> str:
     delays = [5, 10, 20]
@@ -69,11 +80,11 @@ def build_single_system_context(query: str, max_total_chars: int = 9000, k: int
         snippets.append(txt)
 
     if not snippets:
-        return ("You are a strict RAG assistant. No context was retrieved from the vector store for this query. "
-                "If the answer is not present, say you don’t know.")
+        return ("You are ghaymah expert . follow instraction to be strict RAG assistant. No context was retrieved from the vector store for this query. "
+                "If the answer is not present, say do not mention in ghaymah documentation.")
 
-    header = ("You are a strict RAG assistant. Answer ONLY using the provided context snippets. "
-              "If the answer is not present, say you don’t know. ")
+    header = ("You are ghaymah expert. follow instraction to be strict RAG assistant. Answer ONLY using the provided context snippets. "
+              "If the answer is not present, say do not mention in ghaymah documentation.")
     body_budget = max_total_chars - len(header)
     body_parts, used = [], 0
     for snip in snippets:
@@ -89,7 +100,7 @@ def build_single_system_context(query: str, max_total_chars: int = 9000, k: int
     footer = "Sources:\n" + "\n".join(f"- {s}" for s in uniq_sources) + "\n" if uniq_sources else ""
     return (header + "".join(body_parts) + footer).strip()
 
-SYSTEM_SEED = "You are a strict RAG assistant. Answer ONLY using the provided context."
+SYSTEM_SEED = "You are ghaymah expert. follow instraction to be strict RAG assistant. Answer ONLY using the provided context snippets."
 def init_state():
     return {"messages": [{"role": "system", "content": SYSTEM_SEED}], "last_call_ts": None}
 
@@ -101,7 +112,6 @@ def record_call_time(state: dict):
     state["last_call_ts"] = time.time()
 
 def respond(user_message: str, state: dict):
-    # Basic env checks – we still show a bot response so the UI proves it’s working
     missing = []
     if not API_KEY: missing.append("API_KEY")
     if not HOST: missing.append("HOST")
@@ -132,19 +142,27 @@ def respond(user_message: str, state: dict):
     state["messages"].append({"role": "assistant", "content": reply})
     return reply, state
 
-#  Gradio UI: messages API (Gradio >= 5)
-with gr.Blocks(title="Ghaymah Chatbot") as demo:
-    state = gr.State(init_state())  # {"messages": [...], "last_call_ts": ...}
+#  Gradio UI
+with gr.Blocks(title=f"{APP_Name} v{APP_Version}", css=CSS) as demo:
+    header_logo_src = logo_data_uri(COMPANY_LOGO)
+    logo_html = f"<img src='{header_logo_src}' alt='logo'>" if header_logo_src else ""
+    gr.HTML(f"""
+    <div class="app-header">
+        {logo_html}
+        <div class="app-header-text">
+            <div class="app-title">{APP_Name}</div>
+            <div class="app-sub">v{APP_Version}  •  {OWNER_NAME}</div>
+        </div>
+    </div>
+    """)
+    state = gr.State(init_state())  
 
     with gr.Row():
         # LEFT: chat + input
         with gr.Column(scale=3):
-            gr.Markdown("# ☁ Ghaymah Chatbot")
-
-            # Start with an explicit empty list so it's never None
+        
             chatbot = gr.Chatbot(label="Chat", height=520, type="messages", value=[])
-
-            # Input + buttons under the chat
+            
             txt = gr.Textbox(
                 placeholder="Ask anything about the Ghaymah documentation…",
                 label="Your message",
@@ -155,13 +173,13 @@ with gr.Blocks(title="Ghaymah Chatbot") as demo:
                 send_btn = gr.Button("Send", variant="primary")
                 clear_btn = gr.Button("Clear")
 
-        # RIGHT: logo + status text (from the first screenshot), stacked vertically
+        # RIGHT
         with gr.Column(scale=1, min_width=300):
             gr.Image(
                 value="download.jpeg",
                 interactive=False,
                 show_label=False,
-                container=False,                 # no surrounding card/frame
+                container=False,                
                 show_fullscreen_button=False,
             )
             gr.Markdown(
@@ -170,24 +188,21 @@ with gr.Blocks(title="Ghaymah Chatbot") as demo:
                 f"RPM limit: **{RPM_LIMIT}** (min {MIN_SECONDS_BETWEEN}s between calls)  \n"
             )
 
-    # Step 1: add a user message immediately
     def _on_user_submit(user_input, chat_messages):
         try:
             if not user_input:
                 return "", (chat_messages or [])
-            chat_messages = chat_messages or []  # guard for None
+            chat_messages = chat_messages or []  
             updated = chat_messages + [{"role": "user", "content": user_input}]
             print("[on_submit] user:", user_input)
             return "", updated
         except Exception as e:
             print("[on_submit][ERROR]", repr(e))
-            # keep textbox text so you can retry; don't mutate chat on error
             return user_input, (chat_messages or [])
 
     txt.submit(_on_user_submit, [txt, chatbot], [txt, chatbot])
     send_btn.click(_on_user_submit, [txt, chatbot], [txt, chatbot])
 
-    # Step 2: call backend and append assistant message
     def _bot_step(chat_messages, state):
         try:
             chat_messages = chat_messages or []
@@ -201,24 +216,21 @@ with gr.Blocks(title="Ghaymah Chatbot") as demo:
                 return chat_messages, state
 
             print("[bot_step] responding to:", last_user)
-            bot_reply, new_state = respond(last_user, state)  # <-- your 2-arg respond
+            bot_reply, new_state = respond(last_user, state)  
 
             updated = chat_messages + [{"role": "assistant", "content": bot_reply}]
             return updated, new_state
 
         except Exception as e:
             print("[bot_step][ERROR]", repr(e))
-            # show the error in the chat so you see *something* in the UI
             updated = (chat_messages or []) + [
                 {"role": "assistant", "content": f"⚠️ Internal error: {e}"}
             ]
             return updated, state
 
-    # Submit (Enter)
     txt.submit(_on_user_submit, [txt, chatbot], [txt, chatbot])\
         .then(_bot_step, [chatbot, state], [chatbot, state])
 
-    # Click (Send)
     send_btn.click(_on_user_submit, [txt, chatbot], [txt, chatbot])\
         .then(_bot_step, [chatbot, state], [chatbot, state])