Update app.py

app.py

@@ -15,22 +15,47 @@ def generate(prompt, max_new_tokens):
     output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
     return output[0]
 
-# Streamlit App Layout
-st.markdown(f"<img src='{logo_url}' style='max-height: 100px;'>", unsafe_allow_html=True)  # Logo einfügen
-st.title("ChatGPT_Streamlit-Prompt Generator")
-st.write("Gib eine Rolle ein, und es wird ein Prompt basierend darauf generiert.")
-prompt = st.text_input("Gib eine Rolle ein, Beispiel: Virtueller Assistent", placeholder="Text hier", value="")
-max_new_tokens = st.slider("Wähle die maximale Anzahl an Tokens in der Antwort", min_value=100, max_value=500, value=150, step=10)
-
-if st.button("Generieren"):
-    output = generate(prompt, max_new_tokens)
-    st.write("Generierter Prompt:")
-    st.write(output)
-
-# Beispiele anzeigen
-st.write("Beispiele:")
-with open("examples.txt", "r") as f:  # Ersetze "examples.txt" mit dem tatsächlichen Pfad zu Deiner Beispieldatei
-    examples = f.readlines()
-random_examples = random.sample(examples, 5)
-for example in random_examples:
-    st.write(f"• {example.strip()}")
+ERLAUBTE_BENUTZER = {
+    "user1": "passwort1",
+    "user2": "passwort2"
+}
+
+def check_login(username, password):
+    """Überprüft die Anmeldedaten."""
+    return username in ERLAUBTE_BENUTZER and password == ERLAUBTE_BENUTZER[username]
+
+def main_page():
+    """Hauptseite der Anwendung, die nach erfolgreicher Anmeldung angezeigt wird."""
+    st.markdown(f"<img src='{logo_url}' style='max-height: 100px;'>", unsafe_allow_html=True)  # Logo einfügen
+    st.title("ChatGPT_Streamlit-Prompt Generator")
+    st.write("Gib eine Rolle ein, und es wird ein Prompt basierend darauf generiert.")
+    prompt = st.text_input("Gib eine Rolle ein, Beispiel: Virtueller Assistent", placeholder="Text hier", value="")
+    max_new_tokens = st.slider("Wähle die maximale Anzahl an Tokens in der Antwort", min_value=100, max_value=500, value=150, step=10)
+    
+    if st.button("Generieren"):
+        output = generate(prompt, max_new_tokens)
+        st.write("Generierter Prompt:")
+        st.write(output)
+
+    # Beispiele anzeigen
+    st.write("Beispiele:")
+    with open("examples.txt", "r") as f:  # Ersetze "examples.txt" mit dem tatsächlichen Pfad zu Deiner Beispieldatei
+        examples = f.readlines()
+    random_examples = random.sample(examples, 5)
+    for example in random_examples:
+        st.write(f"• {example.strip()}")
+
+def login_page():
+    """Anmeldeseite."""
+    st.sidebar.title("Anmeldung")
+    username = st.sidebar.text_input("Benutzername")
+    password = st.sidebar.text_input("Passwort", type="password")
+    if st.sidebar.button("Anmelden"):
+        if check_login(username, password):
+            # Erfolgreiche Anmeldung, zeigt die Hauptseite an
+            main_page()
+        else:
+            st.sidebar.error("Fehler: Falscher Benutzername oder Passwort.")
+
+if __name__ == "__main__":
+    login_page()