Update app.py

app.py

@@ -2,82 +2,98 @@ import os
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-def generate_novel_chapter(genre, style, premise, tone, length_tokens, previous_story=""):
-    print(f"--- GENERIERUNG GESTARTET: Genre={genre} ---")
-    
-    if not premise or not premise.strip():
-        raise gr.Error("Bitte gib eine Handlungsvorgabe ein.")
+# Einfacher In-Memory-Speicher pro User
+USER_STORIES = {}
 
-    HF_TOKEN = os.getenv("HF_TOKEN")
-    if not HF_TOKEN:
-        raise gr.Error("HF_TOKEN ist nicht in den Space-Secrets hinterlegt.")
+def generate_novel_chapter(genre, style, premise, tone, length_tokens, previous_story="", request: gr.Request = None):
+    username = "Gast"
+    if request and request.query_params:
+        username = request.query_params.get("username", "Gast")
+    
+    print(f"--- GENERIERUNG FÜR USER: {username} ---")
+    
+    if not premise or not premise.strip():
+        raise gr.Error("Bitte gib eine Handlungsvorgabe ein.")
 
-    try:
-        client = InferenceClient(token=HF_TOKEN)
+    HF_TOKEN = os.getenv("HF_TOKEN")
+    if not HF_TOKEN:
+        raise gr.Error("HF_TOKEN ist nicht in den Space-Secrets hinterlegt.")
 
-        system_instruction = (
-            f"Du bist ein preisgekrönter Romanautor. Genre: {genre}. Stil: {style}. Ton: {tone}. "
-            "WICHTIGE ANWEISUNG: Schreibe das gesamte Kapitel AUSSCHLIESSLICH auf DEUTSCH."
-        )
+    try:
+        client = InferenceClient(token=HF_TOKEN)
 
-        # Wenn es bereits eine vorherige Geschichte gibt, bauen wir eine Fortsetzung an
-        if previous_story and previous_story.strip():
-            user_prompt = (
-                f"Hier ist die bisherige Geschichte (Auszug/Zusammenfassung der letzten Teile):\n{previous_story[-1000:]}\n\n"
-                f"Schreibe basierend darauf das NÄCHSTE Kapitel nahtlos weiter. Die neue Handlungsvorgabe für dieses Kapitel ist: {premise}"
-            )
-        else:
-            user_prompt = f"Schreibe ein detailliertes erstes Kapitel basierend auf: {premise}"
+        system_instruction = (
+            f"Du bist ein preisgekrönter Romanautor. Genre: {genre}. Stil: {style}. Ton: {tone}. "
+            "WICHTIGE ANWEISUNG: Schreibe das gesamte Kapitel AUSSCHLIESSLICH auf DEUTSCH."
+        )
 
-        response = client.chat_completion(
-            model="Qwen/Qwen2.5-72B-Instruct",
-            messages=[
-                {"role": "system", "content": system_instruction},
-                {"role": "user", "content": user_prompt}
-            ],
-            max_tokens=int(length_tokens),
-            temperature=0.7,
-            stream=False 
-        )
+        if previous_story and previous_story.strip():
+            user_prompt = (
+                f"Hier ist die bisherige Geschichte (Auszug/Zusammenfassung der letzten Teile):\n{previous_story[-1000:]}\n\n"
+                f"Schreibe basierend darauf das NÄCHSTE Kapitel nahtlos weiter. Die neue Handlungsvorgabe für dieses Kapitel ist: {premise}"
+            )
+        else:
+            user_prompt = f"Schreibe ein detailliertes erstes Kapitel basierend auf: {premise}"
 
-        full_text = response.choices[0].message.content
-        print("--- KAPITEL ERFOLGREICH GENERIERT ---")
-        
-        # Wir geben das neue Kapitel zurück und hängen es an die Gesamthistorie an
-        updated_story = (previous_story + "\n\n--- NÄCHSTES KAPITEL ---\n\n" + full_text) if previous_story else full_text
-        return updated_story, updated_story
+        response = client.chat_completion(
+            model="Qwen/Qwen2.5-72B-Instruct",
+            messages=[
+                {"role": "system", "content": system_instruction},
+                {"role": "user", "content": user_prompt}
+            ],
+            max_tokens=int(length_tokens),
+            temperature=0.7,
+            stream=False 
+        )
 
-    except Exception as e:
-        print(f"--- FEHLER: {str(e)} ---")
-        raise gr.Error(f"Fehler: {str(e)}")
+        full_text = response.choices[0].message.content
+        
+        updated_story = (previous_story + "\n\n--- NÄCHSTES KAPITEL ---\n\n" + full_text) if previous_story else full_text
+        USER_STORIES[username] = updated_story
+        
+        return updated_story, updated_story
+
+    except Exception as e:
+        print(f"--- FEHLER: {str(e)} ---")
+        raise gr.Error(f"Fehler: {str(e)}")
+
+def load_user_session(request: gr.Request = None):
+    username = "Gast"
+    if request and request.query_params:
+        username = request.query_params.get("username", "Gast")
+    
+    welcoming = f"👋 Hallo **{username}**! Deine Schreibsession ist bereit."
+    existing_story = USER_STORIES.get(username, "")
+    return welcoming, existing_story, existing_story
 
 # Interface
 with gr.Blocks() as demo:
-    gr.Markdown("# 📚 GamerJam.de AI Story Studio")
-    
-    # Zustand, um die gesamte bisherige Story im Hintergrund zu speichern
-    story_history_state = gr.State("")
+    welcome_message = gr.Markdown("Lade Benutzerdaten...")
+    story_history_state = gr.State("")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            genre = gr.Dropdown(["High Fantasy", "Sci-Fi", "Cyberpunk", "Abenteuer", "Drama", "Krimi", "Thriller"], label="Genre", value="High Fantasy")
+            style = gr.Dropdown(["Erzähler (3. Person)", "Ich-Perspektive"], label="Erzählstil", value="Erzähler (3. Person)")
+            tone = gr.Dropdown(["Düster", "Episch", "Spannend", "Poetisch", "Humorvoll"], label="Tonalität", value="Episch")
+            length = gr.Slider(500, 2500, value=1200, label="Tokens pro Kapitel")
+        with gr.Column(scale=2):
+            premise = gr.Textbox(label="Handlungsvorgabe", lines=6, placeholder="Gib hier ein, was im nächsten Kapitel passieren soll...")
+            submit_btn = gr.Button("Kapitel generieren / fortsetzen", variant="primary")
+            
+    output = gr.Textbox(label="Dein fortlaufendes Buch", lines=15)
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            genre = gr.Dropdown(["High Fantasy", "Sci-Fi", "Cyberpunk", "Abenteuer", "Drama", "Krimi", "Thriller"], label="Genre", value="High Fantasy")
-            style = gr.Dropdown(["Erzähler (3. Person)", "Ich-Perspektive"], label="Erzählstil", value="Erzähler (3. Person)")
-            tone = gr.Dropdown(["Düster", "Episch", "Spannend", "Poetisch", "Humorvoll"], label="Tonalität", value="Episch")
-            length = gr.Slider(500, 2500, value=1200, label="Tokens pro Kapitel")
-        with gr.Column(scale=2):
-            premise = gr.Textbox(label="Handlungsvorgabe (für den nächsten Schritt / das nächste Kapitel)", lines=6, placeholder="Gib hier ein, was im nächsten Kapitel passieren soll...")
-            
-            with gr.Row():
-                submit_btn = gr.Button("Kapitel generieren / fortsetzen", variant="primary")
-            
-    output = gr.Textbox(label="Dein fortlaufendes Buch", lines=15)
-    
-    # Klick-Event verknüpft die Eingaben und aktualisiert sowohl das Ausgabefeld als auch den State
-    submit_btn.click(
-        fn=generate_novel_chapter, 
-        inputs=[genre, style, premise, tone, length, story_history_state], 
-        outputs=[output, story_history_state]
-    )
+    demo.load(
+        fn=load_user_session,
+        inputs=None,
+        outputs=[welcome_message, output, story_history_state]
+    )
+    
+    submit_btn.click(
+        fn=generate_novel_chapter, 
+        inputs=[genre, style, premise, tone, length, story_history_state], 
+        outputs=[output, story_history_state]
+    )
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Default())
+    demo.launch(theme=gr.themes.Default())