Update app.py

app.py

@@ -1,6 +1,4 @@
-# app.py — Insurance Q&A (RAG) with Omantel logo from GitHub URL (centered top)
-# Minimal changes; logic preserved. Uses Pinecone + LlamaIndex + OpenAI.
-
+# app.py — Omantel Insurance Q&A (RAG) with local logo at top-center
 import os
 import logging
 import gradio as gr
@@ -22,15 +20,17 @@ EMBED_MODEL         = os.getenv("EMBED_MODEL", "text-embedding-3-small")  # 1536
 LLM_MODEL           = os.getenv("LLM_MODEL", "gpt-4o-mini")
 
 DATA_DIR = "data"
-DEFAULT_TOP_K = 4  # internal similarity_top_k (no UI control)
+DEFAULT_TOP_K = 4  # internal similarity_top_k
 
-# Omantel logo (raw GitHub URL so it renders directly)
-LOGO_URL = "https://raw.githubusercontent.com/Decoding-Data-Science/Omantel/main/Omantel_Logo%20(1).png"
+# ---- Local logo (commit this image to your Space repo) ----
+LOGO_PATH = os.path.join(DATA_DIR, "Omantel_Logo_new.png")
 
 if not PINECONE_API_KEY:
-    raise RuntimeError("Missing PINECONE_API_KEY (set it in your Space → Settings → Variables).")
+    raise RuntimeError("Missing PINECONE_API_KEY (Space → Settings → Variables).")
 if not OPENAI_API_KEY:
-    raise RuntimeError("Missing OPENAI_API_KEY (set it in your Space → Settings → Variables).")
+    raise RuntimeError("Missing OPENAI_API_KEY (Space → Settings → Variables).")
+if not os.path.exists(LOGO_PATH):
+    raise RuntimeError("Logo not found: data/Omantel_Logo_new.png (commit it to your Space repo).")
 
 logging.basicConfig(level=logging.INFO)
 log = logging.getLogger("dds-space")
@@ -59,14 +59,11 @@ vector_store = PineconeVectorStore(pinecone_index=pinecone_index)
 def bootstrap_index():
     if not os.path.isdir(DATA_DIR):
         raise RuntimeError("No 'data/' directory found. Commit your documents to data/ in the Space repo.")
-    log.info("Loading documents from ./data ...")
     docs = SimpleDirectoryReader(DATA_DIR).load_data()
     if not docs:
         raise RuntimeError("No documents found in data/. Add e.g., data/insurance.pdf")
-    log.info(f"Docs loaded: {len(docs)}. Upserting into Pinecone…")
     storage_ctx = StorageContext.from_defaults(vector_store=vector_store)
     VectorStoreIndex.from_documents(docs, storage_context=storage_ctx, show_progress=True)
-    log.info("Index upsert complete.")
 
 bootstrap_index()
 
@@ -98,23 +95,20 @@ def use_faq(selected_faq: str, free_text: str):
 
 # ===== UI =====
 CSS = """
-.header { text-align:center; }
-.header img { max-height:80px; height:auto; }
+.header { display:flex; flex-direction:column; align-items:center; gap:6px; }
+.header .logo { max-height:80px; width:auto; }
 .title { text-align:center; font-weight:700; font-size:1.4rem; margin:6px 0 0 0; }
 .subnote { text-align:center; margin-top:-2px; opacity:0.8; }
 """
 
 with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
-    # Centered logo + title
-    gr.Markdown(
-        f"""
-<div class="header">
-  <img src="{LOGO_URL}" alt="Omantel logo" />
-</div>
-<h1 class="title">Omantel Insurance Q&A — RAG Assistant</h1>
-<p class="subnote">Ask about coverage, claims, exclusions, and more — powered by LlamaIndex + Pinecone</p>
-"""
-    )
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("<div class='header'>")
+            gr.Image(value=LOGO_PATH, show_label=False, elem_classes=["logo"])
+            gr.Markdown("<h1 class='title'>Omantel Insurance Q&A — AI Assistant</h1>"
+                        "<p class='subnote'>Ask about coverage, claims, exclusions, and more — powered by LlamaIndex + Pinecone</p>")
+            gr.Markdown("</div>")
 
     with gr.Row():
         with gr.Column(scale=1):