Update app.py

agenten eingeführt

app.py

@@ -1,12 +1,36 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
+import torch
 import PyPDF2
-import datetime
 
-# Initialize the Hugging Face text generation pipeline
-model = pipeline('text-generation', model='distilgpt2', framework='pt')
+# Initialisiere die Text-Generierungs-Pipeline für verschiedene Kapitel
+kapitel_agenten = {
+    "Einführung in die weibliche Anatomie": pipeline('text-generation', model='distilgpt2', framework='pt'),
+    "Der Menstruationszyklus": pipeline('text-generation', model='distilgpt2', framework='pt'),
+    "Ernährung und Lebensstil": pipeline('text-generation', model='distilgpt2', framework='pt'),
+    "Stressmanagement und Schlaf": pipeline('text-generation', model='distilgpt2', framework='pt'),
+    "Umweltgifte und hormonelle Gesundheit": pipeline('text-generation', model='distilgpt2', framework='pt'),
+    "Empfohlene Tees und Kräuter": pipeline('text-generation', model='distilgpt2', framework='pt'),
+    "Weitere Tipps und Strategien": pipeline('text-generation', model='distilgpt2', framework='pt'),
+    "Zusammenfassung und Ausblick": pipeline('text-generation', model='distilgpt2', framework='pt')
+}
 
-# Load and parse the PDF file for the knowledge base
+# Intent-Detection-Modell initialisieren
+intent_model_name = "bert-base-uncased"
+intent_model = AutoModelForSequenceClassification.from_pretrained(intent_model_name)
+intent_tokenizer = AutoTokenizer.from_pretrained(intent_model_name)
+
+intents = ["Anatomie", "Zyklus", "Ernährung", "Stress", "Umweltgifte", "Tees", "Tipps", "Zusammenfassung"]
+
+# Funktion zur Erkennung der Benutzerintention
+def erkenne_intention(text):
+    inputs = intent_tokenizer(text, return_tensors="pt", truncation=True, padding=True)
+    outputs = intent_model(**inputs)
+    probs = torch.nn.functional.softmax(outputs.logits, dim=-1)
+    intent_idx = torch.argmax(probs, dim=1).item()
+    return intents[intent_idx]
+
+# PDF-Dokument laden und verarbeiten
 pdf_path = "Einsteiger Guide Leben im Einklang mit dem weiblichen Zyklus.pdf"
 pdf_reader = PyPDF2.PdfReader(pdf_path)
 pdf_text = ""
@@ -14,15 +38,27 @@ for page_num in range(len(pdf_reader.pages)):
     page = pdf_reader.pages[page_num]
     pdf_text += page.extract_text()
 
-# Define the CSS style for the UI with a mobile-friendly design
+# Define the CSS style for the UI with a mobile-friendly design and centered content
 st.markdown("""
     <style>
     body {
         font-family: 'Satoshi', sans-serif;
         background-color: white;
+        display: flex;
+        justify-content: center;
+        align-items: center;
+        text-align: center;
+        padding: 0;
+        margin: 0;
     }
     .main {
         background-color: white;
+        display: flex;
+        justify-content: center;
+        align-items: center;
+        flex-direction: column;
+        width: 100%;
+        padding: 2rem;
     }
     .stButton>button {
         background-color: #ff4d94;
@@ -32,10 +68,14 @@ st.markdown("""
     .stTextInput>div>div>input {
         background-color: #ff99c2;
         color: white;
+        width: 100%;
+        text-align: center;
     }
     .stTextArea>div>textarea {
         background-color: #ff99c2;
         color: white;
+        width: 100%;
+        text-align: center;
     }
     .stSubheader, .stText, .stHeader, h1, h2 {
         color: #b30047;
@@ -62,69 +102,39 @@ st.markdown("""
     """, unsafe_allow_html=True)
 
 # Set title and subtitle
-st.title("Welcome to Your Menstrual Cycle Tracker")
-st.subheader("Align with Your Body")
+st.title("Willkommen bei Ihrem Menstruationszyklus-Assistenten")
+st.subheader("Ein Assistent für Business Frauen")
 
-# Input fields for user parameters
-days_menstruation = st.number_input("Days of Menstruation", min_value=1, max_value=10, step=1)
-days_cycle = st.number_input("Days of Cycle", min_value=21, max_value=35, step=1)
-last_period_start_date = st.date_input("Last Period Start Date")
-mood_past_3_days = st.text_area("Mood of the Past 3 Days")
-goals_next_3_months = st.text_area("Goals for the Next 3 Months")
-
-# Function to determine the current cycle phase
-def determine_cycle_phase(days_menstruation, days_cycle, last_period_start_date):
-    today = datetime.date.today()
-    cycle_day = (today - last_period_start_date).days % days_cycle
-
-    if cycle_day < days_menstruation:
-        return "Menstrual Phase (Winter)"
-    elif cycle_day < days_menstruation + (days_cycle - days_menstruation) // 2:
-        return "Follicular Phase (Spring)"
-    elif cycle_day < days_cycle - (days_cycle - days_menstruation) // 2:
-        return "Ovulation Phase (Summer)"
-    else:
-        return "Luteal Phase (Autumn)"
+# Eingabefelder für Benutzerparameter
+antworten = st.text_area("Beschreiben Sie Ihr Anliegen oder Ihre Fragen (z.B. Anatomie, Zyklus, Ernährung, Stress, etc.)")
 
 # Submit button
-if st.button("Tell Me About My Cycle"):
-    if days_menstruation and days_cycle and last_period_start_date:
-        # Determine current cycle phase
-        current_phase = determine_cycle_phase(days_menstruation, days_cycle, last_period_start_date)
-
-        # Fetch relevant information from the PDF
-        phase_info = ""
-        if "Menstruationsphase" in pdf_text and "Ernährung" in pdf_text:
-            phase_info = pdf_text.split("Menstruationsphase")[1].split("Ernährung")[0]
-        else:
-            phase_info = "I am sorry I am only an Assistant and I advise you to call an expert."
-
-        # Display current phase information
-        st.subheader(f"Current Phase: {current_phase}")
-        st.write(phase_info)
-
-        # Knowledge base information
-        st.header("Menstrual Cycle Knowledge Base")
-        st.write("""
-        The menstrual cycle is a natural process that occurs in the female body to prepare for a potential pregnancy. It consists of several phases:
-        - **Menstrual Phase:** This is the phase where the uterine lining sheds, causing bleeding.
-        - **Follicular Phase:** The body prepares for ovulation by growing follicles in the ovaries.
-        - **Ovulation Phase:** An egg is released from the ovary, making it the most fertile phase.
-        - **Luteal Phase:** The body prepares for a possible pregnancy or the next menstrual cycle.
+if st.button("Frage beantworten"):
+    if antworten:
+        # Bestimme die Intention des Benutzers
+        intent = erkenne_intention(antworten)
+        
+        # Bestimme das richtige Kapitel basierend auf der Benutzerintention
+        kapitel = intent  # Direktes Mapping von Intention zu Kapitel
+        
+        # Wählen Sie den richtigen Agenten aus
+        agent = kapitel_agenten[kapitel]
+        
+        # Generiere die Antwort mit dem ausgewählten Agenten
+        input_text = f"Frage: {antworten}\nAntwort:"
+        antwort = agent(input_text, max_new_tokens=150, num_return_sequences=1)[0]['generated_text']
         
-        **Tips for Managing Different Phases:**
-        - **Menstrual Phase:** Rest and engage in gentle activities like reading or journaling.
-        - **Follicular Phase:** Plan new projects and gradually increase your activity.
-        - **Ovulation Phase:** Use your high energy for social and professional networking.
-        - **Luteal Phase:** Focus on completing tasks and integrate stress-reducing activities like yoga.
+        # Display the answer
+        st.subheader(f"Antwort vom Experten für {kapitel}")
+        st.write(antwort)
         
-        Remember, your body and cycle are unique. Embrace and understand your cycle to align better with your body's natural rhythms.
-        """)
+        # Zeige die relevanten Informationen aus dem PDF-Dokument an
+        st.subheader("Menstrual Cycle Knowledge Base")
+        st.write(pdf_text)
     else:
-        st.error("Please fill in all the fields.")
+        st.error("Bitte stellen Sie eine Frage.")
 
-st.header("References")
+st.header("Referenzen")
 st.write("""
-This is a beta-version of 2024-06-02 and we are happy to assist you with the updated version. 
-Greetings Nur M.
+Dies ist eine Beta-Version vom 2024-06-02 und wir freuen uns, Ihnen mit der aktualisierten Version zu helfen. Grüße, Nur M.
 """)