├── requirements.txt
├── .streamlit
    └── .gitignore
├── .devcontainer
    └── devcontainer.json
└── app.py


/requirements.txt:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/d-harshavardhan-sai/AI_STORY_GEN_AI/HEAD/requirements.txt


--------------------------------------------------------------------------------
/.streamlit/.gitignore:
--------------------------------------------------------------------------------
 1 | # Python cache files
 2 | __pycache__/
 3 | *.py[cod]
 4 | *.pyo
 5 | 
 6 | # Virtual environment folders
 7 | .env/
 8 | .venv/
 9 | venv/
10 | ENV/
11 | env/
12 | 
13 | # Jupyter Notebook checkpoints
14 | .ipynb_checkpoints/
15 | 
16 | # System files
17 | .DS_Store
18 | Thumbs.db
19 | 
20 | # VSCode settings (optional)
21 | .vscode/
22 | 
23 | # Streamlit secrets (DO NOT push this to GitHub)
24 | .streamlit/secrets.toml
25 | 
26 | # Logs or temporary files (if any)
27 | *.log
28 | 


--------------------------------------------------------------------------------
/.devcontainer/devcontainer.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "name": "Python 3",
 3 |   // Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile
 4 |   "image": "mcr.microsoft.com/devcontainers/python:1-3.11-bullseye",
 5 |   "customizations": {
 6 |     "codespaces": {
 7 |       "openFiles": [
 8 |         "README.md",
 9 |         "app.py"
10 |       ]
11 |     },
12 |     "vscode": {
13 |       "settings": {},
14 |       "extensions": [
15 |         "ms-python.python",
16 |         "ms-python.vscode-pylance"
17 |       ]
18 |     }
19 |   },
20 |   "updateContentCommand": "[ -f packages.txt ] && sudo apt update && sudo apt upgrade -y && sudo xargs apt install -y <packages.txt; [ -f requirements.txt ] && pip3 install --user -r requirements.txt; pip3 install --user streamlit; echo '✅ Packages installed and Requirements met'",
21 |   "postAttachCommand": {
22 |     "server": "streamlit run app.py --server.enableCORS false --server.enableXsrfProtection false"
23 |   },
24 |   "portsAttributes": {
25 |     "8501": {
26 |       "label": "Application",
27 |       "onAutoForward": "openPreview"
28 |     }
29 |   },
30 |   "forwardPorts": [
31 |     8501
32 |   ]
33 | }


--------------------------------------------------------------------------------
/app.py:
--------------------------------------------------------------------------------
 1 | import os
 2 | import streamlit as st
 3 | from huggingface_hub import InferenceClient
 4 | 
 5 | # Set HuggingFace token
 6 | HF_TOKEN = st.secrets["HF_TOKEN"]
 7 | 
 8 | client = InferenceClient(api_key=HF_TOKEN)
 9 | 
10 | def generate_story(prompt):
11 |     system_message = "You are a creative writing assistant."
12 |     user_message = f"Write a short, imaginative story based on this prompt: {prompt}"
13 |     messages = [
14 |         {"role": "system", "content": system_message},
15 |         {"role": "user", "content": user_message}
16 |     ]
17 |     response = client.chat.completions.create(
18 |         model="meta-llama/Meta-Llama-3.1-8B-Instruct",
19 |         messages=messages,
20 |         max_tokens=512,
21 |         temperature=0.8,
22 |     )
23 |     return response.choices[0].message.content.strip()
24 | 
25 | # Streamlit UI
26 | st.set_page_config(page_title="AI Story Generator", layout="centered")
27 | st.title("📚 AI Story Generator from Prompts")
28 | st.markdown("Enter a creative prompt and let AI write a story for you!")
29 | 
30 | prompt = st.text_area("✍️ Enter your story prompt:", height=150)
31 | 
32 | if st.button("🚀 Generate Story"):
33 |     if not prompt.strip():
34 |         st.warning("⚠️ Please enter a story prompt.")
35 |     else:
36 |         with st.spinner("🧠 Generating..."):
37 |             try:
38 |                 story = generate_story(prompt)
39 |                 st.success("🎉 Here's your story:")
40 |                 st.write(story)
41 |             except Exception as e:
42 |                 st.error(f"❌ Error: {e}")
43 | 


--------------------------------------------------------------------------------