Upload 15 files

.gitattributes

@@ -1,35 +1,2 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.py linguist-language=Python
+*.ipynb -text

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Himanshu Kumar
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,3 +1,21 @@
----
-license: apache-2.0
----
+---
+language:
+  - en
+license: mit
+tags:
+  - text-generation
+  - worldbuilding
+  - creative-ai
+  - storytelling
+pipeline_tag: text-generation
+library_name: transformers
+model_name: dreamforge
+base_model: google/flan-t5-xl
+---
+
+# 🌌 DREAMFORGE — Generative World Builder
+
+DREAMFORGE converts short seed prompts into rich fictional universes (world names, geography, species, histories, factions, culture, and story hooks).
+This repo contains a lightweight pipeline and a Streamlit UI to experiment locally.
+
+See `examples/` for sample prompts and `notebooks/` for a demo notebook.

api.py

@@ -0,0 +1,15 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from world_generator import generate_world
+from utils import load_config
+
+app = FastAPI(title='DREAMFORGE API')
+
+class Request(BaseModel):
+    prompt: str
+
+@app.post('/generate')
+def generate(req: Request):
+    cfg = load_config()
+    text = generate_world(req.prompt)
+    return {'world': text}

app.py

@@ -0,0 +1,30 @@
+import streamlit as st
+from world_generator import generate_world
+from lore_generator import generate_lore
+from species_generator import generate_species
+from utils import load_config
+
+st.set_page_config(page_title='DREAMFORGE', layout='wide')
+st.title('🌌 DREAMFORGE — Generative World Builder')
+
+cfg = load_config()
+
+seed = st.text_area('Seed prompt', value='A mist-covered archipelago where floating cities harvest storms.')
+extra = st.text_area('Extra notes (optional)', value='Low technology, focus on trade and storms.')
+
+if st.button('Generate World'):
+    with st.spinner('Generating world (model may be downloaded on first run)...'):
+        world = generate_world(seed)
+        lore = generate_lore(seed)
+        species = generate_species(seed)
+
+    st.subheader('World Overview')
+    st.markdown(world)
+
+    st.subheader('Lore & Timeline')
+    st.markdown(lore)
+
+    st.subheader('Species & Ecosystems')
+    st.markdown(species)
+
+    st.download_button('Download world (md)', world, file_name='dreamforge_world.md')

config.yaml

@@ -0,0 +1,13 @@
+generation:
+  model_name: "google/flan-t5-xl"
+  max_new_tokens: 600
+  num_beams: 4
+  temperature: 0.85
+  top_p: 0.9
+templates:
+  world_overview: |
+    Generate a detailed fictional world from the prompt below.
+    Provide sections: World Name, Summary (1 paragraph), Geography (regions, climate), Major Species (brief descriptions),
+    Key Locations (3-6 with short descriptions), History (timeline: 5 events), Factions & Politics (major groups),
+    Culture & Technology (short notes), Notable Figures (3 with short bios), Plot Hooks (5 story seeds).
+    Output in clear markdown sections and keep language creative.

examples/prompts.txt

@@ -0,0 +1,5 @@
+a mist-covered archipelago where floating cities harvest storms
+a vast desert where ruins sing at night, inhabited by sand-people
+a planet of bioluminescent jungles and slow-moving sky-whales
+an iceball world with subterranean magma gardens and glass-harvesters
+an industrial renaissance island ruled by guilds of tinkers and wind-sailors

examples/sample_output.md

@@ -0,0 +1,7 @@
+# Example: a mist-covered archipelago where floating cities harvest storms
+
+## World Name
+The Stormforges
+
+## Summary
+A chain of islands shrouded in perpetual mist, each crowned with floating cities that harvest and channel storms for energy...

lore_generator.py

@@ -0,0 +1,12 @@
+import torch
+from transformers import pipeline
+from utils import load_config
+
+def generate_lore(seed_prompt):
+    cfg = load_config()
+    model_name = cfg['generation']['model_name']
+    device = 0 if torch.cuda.is_available() else -1
+    pipe = pipeline('text2text-generation', model=model_name, device=device)
+    prompt = f"Provide a deep history and mythology for this world seed:\n{seed_prompt}\n\nRespond in markdown with timeline and myth summaries."
+    out = pipe(prompt, max_new_tokens=400, do_sample=True, temperature=cfg['generation']['temperature'])
+    return out[0].get('generated_text') if isinstance(out, list) else str(out)

notebooks/demo_dreamforge.ipynb

@@ -0,0 +1 @@
+{"cells": [{"cell_type": "markdown", "metadata": {}, "source": ["# DREAMFORGE demo", "Run world_generator.py from the notebook or CLI."]}, {"cell_type": "code", "metadata": {}, "source": ["!python world_generator.py --prompt \"a frozen desert ruled by glass cities\" --out demo_world.md\n", "!sed -n '1,200p' demo_world.md\n"]}], "metadata": {"kernelspec": {"display_name": "Python 3", "language": "python", "name": "python3"}, "language_info": {"name": "python"}}, "nbformat": 4, "nbformat_minor": 5}

prompt_builder.py

@@ -0,0 +1,11 @@
+from utils import load_config
+import textwrap
+
+def build_world_prompt(user_prompt: str, extra_notes: str = "") -> str:
+    cfg = load_config()
+    template = cfg.get('templates', {}).get('world_overview', '').strip()
+    parts = ["User seed:", user_prompt.strip()]
+    if extra_notes:
+        parts += ["Extra notes:", extra_notes.strip()]
+    parts += ["Instructions:", template]
+    return "\n\n".join(parts)

requirements.txt

@@ -0,0 +1,9 @@
+transformers>=4.30.0
+torch>=1.12.0
+accelerate
+sentencepiece
+pyyaml
+streamlit
+fastapi
+uvicorn
+pydantic

species_generator.py

@@ -0,0 +1,12 @@
+import torch
+from transformers import pipeline
+from utils import load_config
+
+def generate_species(seed_prompt):
+    cfg = load_config()
+    model_name = cfg['generation']['model_name']
+    device = 0 if torch.cuda.is_available() else -1
+    pipe = pipeline('text2text-generation', model=model_name, device=device)
+    prompt = f"Invent 4-6 major species for a world described as:\n{seed_prompt}\n\nProvide short bios and ecological roles."
+    out = pipe(prompt, max_new_tokens=300, do_sample=True, temperature=cfg['generation']['temperature'])
+    return out[0].get('generated_text') if isinstance(out, list) else str(out)

utils.py

@@ -0,0 +1,11 @@
+from pathlib import Path
+import yaml
+
+def load_config(path='config.yaml'):
+    p = Path(path)
+    if not p.exists():
+        raise FileNotFoundError(f"Missing config file: {path}")
+    return yaml.safe_load(p.read_text())
+
+def save_text(path, text):
+    Path(path).write_text(text, encoding='utf-8')

world_generator.py

@@ -0,0 +1,34 @@
+import torch
+from transformers import pipeline
+from prompt_builder import build_world_prompt
+from utils import load_config, save_text
+
+def get_pipeline(model_name):
+    device = 0 if torch.cuda.is_available() else -1
+    # Use text2text for Flan-T5 family
+    return pipeline('text2text-generation', model=model_name, device=device, truncation=True)
+
+def generate_world(prompt_seed, out_path=None):
+    cfg = load_config()
+    model_name = cfg['generation']['model_name']
+    max_new_tokens = cfg['generation'].get('max_new_tokens', 512)
+    num_beams = cfg['generation'].get('num_beams', 4)
+    temperature = cfg['generation'].get('temperature', 0.9)
+    top_p = cfg['generation'].get('top_p', 0.95)
+
+    prompt = build_world_prompt(prompt_seed, extra_notes="")
+    pipe = get_pipeline(model_name)
+    result = pipe(prompt, max_new_tokens=max_new_tokens, num_beams=num_beams, do_sample=True, temperature=temperature, top_p=top_p)
+    text = result[0].get('generated_text') if isinstance(result, list) else str(result)
+    if out_path:
+        save_text(out_path, text)
+    return text
+
+if __name__ == '__main__':
+    import argparse
+    p = argparse.ArgumentParser()
+    p.add_argument('--prompt', required=True)
+    p.add_argument('--out', default='dreamforge_output.md')
+    args = p.parse_args()
+    out = generate_world(args.prompt, out_path=args.out)
+    print('Saved to', args.out)