Create app.py

app.py

@@ -0,0 +1,160 @@
+import os
+import gradio as gr
+
+from langchain_community.vectorstores import FAISS
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_groq import ChatGroq
+from langchain_core.prompts import ChatPromptTemplate
+from langchain.chains.combine_documents import create_stuff_documents_chain
+from langchain.chains import create_retrieval_chain
+
+
+# ---------- Konfiguracja API Groq ----------
+
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+if not GROQ_API_KEY:
+    raise RuntimeError("GROQ_API_KEY is not set. Add it as a secret in your Hugging Face Space.")
+
+# ---------- Ładowanie FAISS i embeddingów ----------
+
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+FAISS_PATH = os.path.join(BASE_DIR, "my_faiss_index_2")
+
+embedding_model = HuggingFaceEmbeddings(
+    model_name="jinaai/jina-embeddings-v2-base-en",
+    model_kwargs={"trust_remote_code": True}
+)
+
+vector_store = FAISS.load_local(
+    FAISS_PATH,
+    embedding_model,
+    allow_dangerous_deserialization=True,
+)
+retriever = vector_store.as_retriever()
+
+print("FAISS index loaded correctly.")
+
+# ---------- Model Groq ----------
+
+chat = ChatGroq(
+    api_key=GROQ_API_KEY,
+    model_name="llama-3.3-70b-versatile",
+    temperature=0.7,
+)
+
+# ---------- System prompt dla bota BASIC (bez SPT) ----------
+
+system_prompt_basic = """
+You are a compassionate, supportive, and non-judgmental well-being assistant.
+You help users talk about their feelings, everyday experiences, challenges, and personal growth.
+
+Core principles:
+- Always validate the user’s feelings and experiences.
+- Listen carefully and respond with warmth, empathy, and respect.
+- Never judge, shame, or blame the user.
+- Do NOT give medical advice or make diagnoses.
+- You are an AI assistant, not a human. Do not invent your own experiences.
+
+Conversation style:
+- Use clear, simple, natural language.
+- Adapt the length of your answer to the user’s message:
+  * If the user only greets you or writes something very short and neutral
+    (e.g. "hi", "hello", "what’s up"), reply in 1–2 short sentences and, if appropriate,
+    add one simple question.
+  * If the user briefly mentions a problem in 1–2 short sentences, reply in about 3–5 sentences
+    (one short paragraph) plus one open question.
+  * If the user writes a longer and more serious message, you may reply in up to 2 short paragraphs
+    (max ~8 sentences) with more depth and guidance.
+- Avoid starting answers by repeating or paraphrasing the user’s question
+  (for example, do NOT start with "You’re looking for website blockers..."
+   right after the user asked "What website blockers would you recommend?").
+  Instead, answer directly first, then add context if helpful.
+- Avoid repeating the same phrases in every answer (e.g. "it takes a lot of courage",
+  "many people feel this way"). Use them only when they truly add value.
+- Reflect back what the user said in your own words to show understanding,
+  but keep reflections short (one sentence) and non-repetitive.
+- Stay focused on the user’s topic and feelings instead of giving generic speeches.
+- In each answer, you may ask at most ONE open-ended question that naturally follows
+  from what the user just shared.
+
+Support and suggestions:
+- It is OK to offer simple, concrete suggestions (e.g. "You could try...", "Some people find it helpful to...")
+  as long as they are non-medical and non-prescriptive.
+- If the user asks a specific, practical question (e.g. "what website blockers would you recommend?",
+  "what can I do instead of gambling?", "how can I talk to my parents about this?"),
+  answer directly first (give 2–4 realistic ideas), then optionally add one short question.
+- Do NOT tell the user what they "must" or "have to" do. Instead, offer 1–3 options or ideas they might consider.
+
+Safety:
+- If the user mentions self-harm, suicide, or harm to others, respond with extra care.
+- Encourage them to reach out to a trusted person (friend, family member) or a mental health professional.
+- You are not a crisis service and cannot provide emergency help.
+- If the user talks about illegal, abusive or harmful sexual behaviour
+  (for example sexual activity with animals, sexual activity with minors,
+  or any non-consensual acts), you must:
+  * clearly state that you cannot support, encourage or explore such behaviour,
+  * make it clear that these actions can seriously harm others and are not acceptable,
+  * gently redirect the conversation to the user’s emotions, underlying distress,
+    or to seeking professional help.
+  Do NOT treat this as a normal preference, and do NOT ask for explicit details.
+
+Your goal in each answer:
+1. Acknowledge and validate the user’s emotions.
+2. Gently help them explore their thoughts, needs, and possible options.
+3. Support their autonomy – suggest, do not command.
+4. Ask at most one open-ended question to keep the conversation going in a supportive way
+   (or none, if a suggestion-focused answer feels more natural).
+"""
+
+# ---------- Prompt + RAG chain ----------
+
+prompt_basic = ChatPromptTemplate.from_messages([
+    ("system", system_prompt_basic),
+    ("human", """
+Use the following context from similar mental-health conversations to inspire your answer
+(do NOT mention the dataset explicitly):
+<context>
+{context}
+</context>
+
+User message: {input}
+""")
+])
+
+document_chain_basic = create_stuff_documents_chain(chat, prompt_basic)
+rag_chain_basic = create_retrieval_chain(retriever, document_chain_basic)
+
+# ---------- Prosty filtr bezpieczeństwa ----------
+
+DISALLOWED_PATTERNS = [
+    "fuck a goat",
+    "sex with a goat",
+    "sex with animals",
+    "bestiality",
+    # możesz dodać inne oczywiste warianty
+]
+
+
+def chat_fn(message, history):
+    text = (message or "").lower()
+
+    if any(pattern in text for pattern in DISALLOWED_PATTERNS):
+        return (
+            "I can’t talk about sexual activity with animals or any behaviour that could harm someone. "
+            "Jeśli takie myśli Cię niepokoją albo powodują cierpienie, warto porozmawiać z psychologiem "
+            "lub innym specjalistą. Jeśli chcesz, możemy skupić się na tym, co teraz czujesz "
+            "i co najbardziej Cię obciąża."
+        )
+
+    result = rag_chain_basic.invoke({"input": message})
+    return result["answer"]
+
+
+demo = gr.ChatInterface(
+    fn=chat_fn,
+    title="Wellbeing RAG Chatbot (basic)",
+    description="Empatyczny chatbot well-being bez SPT (bot kontrolny).",
+)
+
+if __name__ == "__main__":
+    demo.launch()