all files added

.gitignore

@@ -0,0 +1,142 @@
+# MY CHNAGES
+secrets.toml
+scratch*.ipynb
+experiments.ipynb
+venv*/
+*.excalidraw
+.vscode*
+.vscode*/
+.examples/
+.swp
+.aws-sam/
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+
+# Environments
+.env
+.venv
+env/
+venv*/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/

0.1

@@ -0,0 +1,34 @@
+Requirement already satisfied: langchain in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (0.0.354)
+Requirement already satisfied: PyYAML>=5.3 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (6.0.2)
+Requirement already satisfied: SQLAlchemy<3,>=1.4 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (2.0.32)
+Requirement already satisfied: aiohttp<4.0.0,>=3.8.3 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (3.10.3)
+Requirement already satisfied: dataclasses-json<0.7,>=0.5.7 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (0.6.7)
+Requirement already satisfied: jsonpatch<2.0,>=1.33 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (1.33)
+Requirement already satisfied: langchain-community<0.1,>=0.0.8 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (0.0.20)
+Requirement already satisfied: langchain-core<0.2,>=0.1.5 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (0.1.23)
+Requirement already satisfied: langsmith<0.1.0,>=0.0.77 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (0.0.87)
+Requirement already satisfied: numpy<2,>=1 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (1.26.4)
+Requirement already satisfied: pydantic<3,>=1 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (2.8.2)
+Requirement already satisfied: requests<3,>=2 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (2.29.0)
+Requirement already satisfied: tenacity<9.0.0,>=8.1.0 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain) (8.5.0)
+Requirement already satisfied: aiohappyeyeballs>=2.3.0 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from aiohttp<4.0.0,>=3.8.3->langchain) (2.3.5)
+Requirement already satisfied: aiosignal>=1.1.2 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from aiohttp<4.0.0,>=3.8.3->langchain) (1.3.1)
+Requirement already satisfied: attrs>=17.3.0 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from aiohttp<4.0.0,>=3.8.3->langchain) (24.2.0)
+Requirement already satisfied: frozenlist>=1.1.1 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from aiohttp<4.0.0,>=3.8.3->langchain) (1.4.1)
+Requirement already satisfied: multidict<7.0,>=4.5 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from aiohttp<4.0.0,>=3.8.3->langchain) (6.0.5)
+Requirement already satisfied: yarl<2.0,>=1.0 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from aiohttp<4.0.0,>=3.8.3->langchain) (1.9.4)
+Requirement already satisfied: marshmallow<4.0.0,>=3.18.0 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from dataclasses-json<0.7,>=0.5.7->langchain) (3.21.3)
+Requirement already satisfied: typing-inspect<1,>=0.4.0 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from dataclasses-json<0.7,>=0.5.7->langchain) (0.9.0)
+Requirement already satisfied: jsonpointer>=1.9 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from jsonpatch<2.0,>=1.33->langchain) (3.0.0)
+Requirement already satisfied: anyio<5,>=3 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain-core<0.2,>=0.1.5->langchain) (4.4.0)
+Requirement already satisfied: packaging<24.0,>=23.2 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from langchain-core<0.2,>=0.1.5->langchain) (23.2)
+Requirement already satisfied: annotated-types>=0.4.0 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from pydantic<3,>=1->langchain) (0.7.0)
+Requirement already satisfied: pydantic-core==2.20.1 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from pydantic<3,>=1->langchain) (2.20.1)
+Requirement already satisfied: typing-extensions>=4.6.1 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from pydantic<3,>=1->langchain) (4.12.2)
+Requirement already satisfied: charset-normalizer<4,>=2 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from requests<3,>=2->langchain) (3.3.2)
+Requirement already satisfied: idna<4,>=2.5 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from requests<3,>=2->langchain) (3.7)
+Requirement already satisfied: urllib3<1.27,>=1.21.1 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from requests<3,>=2->langchain) (1.26.19)
+Requirement already satisfied: certifi>=2017.4.17 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from requests<3,>=2->langchain) (2024.7.4)
+Requirement already satisfied: greenlet!=0.4.17 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from SQLAlchemy<3,>=1.4->langchain) (3.0.3)
+Requirement already satisfied: sniffio>=1.1 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from anyio<5,>=3->langchain-core<0.2,>=0.1.5->langchain) (1.3.1)
+Requirement already satisfied: mypy-extensions>=0.3.0 in c:\users\itsja\appdata\local\programs\python\python311\lib\site-packages (from typing-inspect<1,>=0.4.0->dataclasses-json<0.7,>=0.5.7->langchain) (1.0.0)

README.md

@@ -1,13 +1,62 @@
 ---
-title: Test02
-emoji: 🦀
-colorFrom: pink
-colorTo: yellow
+title: Picture to Story Generator 
+emoji: 📖
+colorFrom: yellow
+colorTo: red
 sdk: gradio
-sdk_version: 4.41.0
+sdk_version: 3.46.0
 app_file: app.py
 pinned: false
-license: apache-2.0
+license: mit
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+<a href="https://huggingface.co/spaces/sssingh/pic-to-story"  target="_blank"><img src="https://img.shields.io/badge/click_here_to_open_gradio_app-orange?style=for-the-badge&logo=dependabot"/></a>
+
+
+# The Storyteller  
+***A Large Language Model Based App to Generate Stories from Pictures***
+
+<img src="https://github.com/sssingh/pic-to-story/blob/main/assets/title.jpg?raw=true" width="1000" height="350"/><br><br> 
+
+>This application employs a Image2Text model hosted by Huggingface, which is a modified adaptation of the Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generation (BLIP) model. First, it generates a descriptive caption for an image. Then, it utilizes this caption to construct a prompt, which is subsequently used with OpenAI's GPT-3.5 to create engaging stories based on the provided picture. 
+
+## App Flow
+
+<img src="https://github.com/sssingh/pic-to-story/blob/main/assets/app-design.png?raw=true" width="1000" height="450"/><br><br> 
+
+BLIP Image2Text model details can be found [here](https://huggingface.co/Sof22/image-caption-large-copy)
+
+## App Details 
+
+* It's important to note that this sample demonstration app is hosted on the free tiers of Huggingface Spaces, which means it is functional but may exhibit slower performance.
+* Additionally, when using the app for the first time or after an extended period (more than 1 hour), you might encounter an "Internal Error" message or receive a story unrelated to the provided image. This is a normal occurrence during the model loading process. Please wait a few seconds and try again; it should function as intended.
+* Please be aware that due to cost and resource constraints, the app currently has a maximum story length limit of 200 words per request.
+
+App UI is shown below:
+
+<img src="https://github.com/sssingh/pic-to-story/blob/main/assets/story-teller-app.png?raw=true" width="1000" height="450"/><br><br> 
+
+**Dark Mode Toggle**: Activate it to switch between dark and light mode.  
+**Image Selector**: Click on it to pick an image from your computer, or drag and drop an image onto it directly. Click the 'X' to clear the selection and resets the app.  
+**Story Genre Dropdown**: Select the desired story genre from the dropdown list.  
+**Story Writing Style Dropdown**: Select the desired story writing style from the dropdown list.  
+**Story Length (in words) Slider**: Adjust the slider to specify the desired length of the generated story.  
+**Creativity Index Slider**: Modify the slider to indicate the desired level of creativity for the generated story. A range between 0.5 and 0.7 is recommended. Setting it to 1.0 results in highly creative, sometimes amusing output.  
+**Generate Story Button**: Press this button to initiate the story generation process.  
+**Clear Button**: Clears all settings and resets the app to its default state.  
+**Story Text Area**: This is where the generated story will be displayed.  
+**Example Section Expander**: Click to expand the section and access built-in examples for quick testing. Simply select an example, click "Generate Story," and no image upload will be necessary.  
+
+The app includes pre-defined examples for your convenience, allowing you to quickly test its capabilities. Explore the examples section, choose one, and click "Generate Story" without needing to upload an image.
+
+<img src="https://github.com/sssingh/pic-to-story/blob/main/assets/story-teller-examples.png?raw=true" width="1000" height="350"/><br><br> 
+
+# Project Source
+[👉 Visit GitHub Repo](https://github.com/sssingh/pic-to-story)
+
+# Contact Me
+[![email](https://img.shields.io/badge/Gmail-D14836?style=for-the-badge&logo=gmail&logoColor=white)](mailto:sunil@sunilssingh.me)
+[![twitter](https://img.shields.io/badge/twitter-1DA1F2?style=for-the-badge&logo=twitter&logoColor=white)](https://twitter.com/@thesssingh)
+[![linkedin](https://img.shields.io/badge/linkedin-0A66C2?style=for-the-badge&logo=linkedin&logoColor=white)](https://www.linkedin.com/in/sssingh/)
+[![website](https://img.shields.io/badge/web_site-8B5BE8?style=for-the-badge&logo=ko-fi&logoColor=white)](https://sunilssingh.me)
+

app.py

@@ -0,0 +1,178 @@
+import gradio as gr
+import model
+from config import app_config
+import mongo_utils as mongo
+
+
+def clear():
+    return None, 50, 0.7, None, None
+
+
+def create_interface():
+    js_enable_darkmode = """() => 
+    {
+        document.querySelector('body').classList.add('dark');
+    }"""
+    js_toggle_darkmode = """() => 
+    {
+        if (document.querySelectorAll('.dark').length) {
+            document.querySelector('body').classList.remove('dark');
+        } else {
+            document.querySelector('body').classList.add('dark');
+        }
+    }"""
+
+    with gr.Blocks(
+        title=app_config.title, theme=app_config.theme, css=app_config.css
+    ) as app:
+        # enable darkmode
+        app.load(fn=None, inputs=None, outputs=None, _js=js_enable_darkmode)
+        with gr.Row():
+            darkmode_checkbox = gr.Checkbox(
+                label="Dark Mode", value=True, interactive=True
+            )
+            # toggle darkmode on/off when checkbox is checked/unchecked
+            darkmode_checkbox.change(
+                None, None, None, _js=js_toggle_darkmode, api_name=False
+            )
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown(
+                    """
+                    # The Storyteller
+                    **This app can craft captivating narratives from captivating images, 
+                    potentially surpassing even Shakespearean standards.  
+                    <br>
+                    Select an `Image` that inspires a story, choose a `Story Genre`, 
+                    `Story Writing Style`, `Story Length (up to 200 words)`, and 
+                    adjust the `Creativity Index` to enhance its creative flair. Then 
+                    hit `Generate Story` button.
+                    Alternatively, just select one the pre-configured `Examples`**  
+                    <br>
+                    Visit the [project's repo](https://github.com/sssingh/pic-to-story)  
+                    <br>
+                    ***Please exercise patience, as the models employed are extensive 
+                    and may require a few seconds to load. If you encounter an unrelated 
+                    story, it is likely still loading; wait a moment and try again.***
+                    """
+                )
+            with gr.Column():
+                max_count = gr.Textbox(
+                    label="Max allowed OpenAI requests:",
+                    value=app_config.openai_max_access_count,
+                )
+                curr_count = gr.Textbox(
+                    label="Used up OpenAI requests:",
+                    value=app_config.openai_curr_access_count,
+                )
+                available_count = gr.Textbox(
+                    label="Available OpenAI requests:",
+                    value=app_config.openai_max_access_count
+                    - app_config.openai_curr_access_count,
+                )
+        with gr.Row():
+            with gr.Column():
+                image = gr.Image(
+                    type="filepath",
+                )
+                with gr.Row():
+                    with gr.Column():
+                        genre = gr.Dropdown(
+                            label="Story Genre: ",
+                            value="Poetry",
+                            choices=app_config.genre,
+                        )
+                        style = gr.Dropdown(
+                            label="Story Writing Style:",
+                            value="Cinematic",
+                            choices=app_config.writing_style_list,
+                        )
+                    with gr.Column():
+                        # Word Count Slider
+                        word_count = gr.Slider(
+                            label="Story Length (words):",
+                            minimum=30,
+                            maximum=200,
+                            value=50,
+                            step=10,
+                        )
+                        creativity = gr.Slider(
+                            label="Creativity Index:",
+                            minimum=0.3,
+                            maximum=1.0,
+                            value=0.7,
+                            step=0.1,
+                        )
+                with gr.Row():
+                    submit_button = gr.Button(
+                        value="Generate Story", elem_classes="orange-button"
+                    )
+                    clear_button = gr.ClearButton(elem_classes="gray-button")
+            with gr.Column():
+                story = gr.Textbox(
+                    label="Story:",
+                    placeholder="Generated story will appear here.",
+                    lines=21,
+                )
+        with gr.Row():
+            with gr.Accordion("Expand for examples:", open=False):
+                gr.Examples(
+                    examples=[
+                        [
+                            "assets/examples/cheetah-deer.jpg",
+                            "Horror",
+                            "Narrative",
+                            80,
+                            0.5,
+                        ],
+                        [
+                            "assets/examples/man-child-pet-dog.jpg",
+                            "Fiction",
+                            "Formal",
+                            100,
+                            0.6,
+                        ],
+                        [
+                            "assets/examples/man-child.jpeg",
+                            "Children Literature",
+                            "Symbolic",
+                            120,
+                            1.0,
+                        ],
+                        [
+                            "assets/examples/men-fighting.jpg",
+                            "Comedy",
+                            "Experimental",
+                            60,
+                            0.6,
+                        ],
+                        [
+                            "assets/examples/teacher-school.jpg",
+                            "Surrealism",
+                            "Non-linear",
+                            100,
+                            0.7,
+                        ],
+                    ],
+                    fn=model.generate_story,
+                    inputs=[image, genre, style, word_count, creativity],
+                    outputs=[story, max_count, curr_count, available_count],
+                    run_on_click=True,
+                )
+        submit_button.click(
+            fn=model.generate_story,
+            inputs=[image, genre, style, word_count, creativity],
+            outputs=[story, max_count, curr_count, available_count],
+        )
+        clear_button.click(
+            fn=clear, inputs=[], outputs=[image, word_count, creativity, story]
+        )
+        image.clear(fn=clear, inputs=[], outputs=[image, word_count, creativity, story])
+
+    return app
+
+
+if __name__ == "__main__":
+    mongo.fetch_curr_access_count()
+    app = create_interface()
+    app.launch()

config.py

@@ -0,0 +1,64 @@
+import os
+from dataclasses import dataclass
+
+
+@dataclass
+class AppConfig:
+    title = "Picture to Story Generator"
+    theme = "freddyaboulton/dracula_revamped"
+    css = "style.css"
+    openai_max_access_count = 200
+    openai_curr_access_count = None
+    mongo_client = None
+    db = "mydb"
+    collection = "pic2story-openai-access-counter"
+    key = "current_count"
+    HF_TOKEN = os.getenv("HF_TOKEN")
+    # OPENAI_KEY = os.getenv("OPENAI_KEY")
+    # I2T_API_URL = os.getenv("I2T_API_URL")
+    # MONGO_CONN_STR = os.getenv("MONGO_CONN_STR")
+    genre_list = genre = [
+        "Adventure",
+        "Children Literature",
+        "Comedy",
+        "Drama",
+        "Fantasy",
+        "Fiction",
+        "Horror",
+        "Mystery",
+        "Non-fiction",
+        "Poetry",
+        "Romance",
+        "Satire",
+        "Surrealism",
+        "Urban Fantasy",
+    ]
+    writing_style_list = [
+        "Cinematic",
+        "Conversational",
+        "Descriptive",
+        "Experimental",
+        "First-Person",
+        "Formal",
+        "Informal",
+        "Metaphorical",
+        "Minimalist",
+        "Narrative",
+        "Non-linear",
+        "Objective",
+        "Sensory",
+        "Stream of Consciousness",
+        "Symbolic",
+        "Third-Person Limited",
+        "Third-Person Omniscient",
+    ]
+    OPENAI_KEY = "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"
+    # I2T_API_URL = (
+    #     "https://api-inference.huggingface.co/models/Sof22/image-caption-large-copy"
+    # )
+    I2T_API_URL = ("https://api-inference.huggingface.co/models/Salesforce/blip2-opt-2.7b")
+    MONGO_CONN_STR = "mongodb+srv://sssingh:Topsycret1@cluster0.fcwxggj.mongodb.net/"
+
+
+
+app_config = AppConfig()

model.py

@@ -0,0 +1,87 @@
+import requests
+from langchain.chat_models import ChatOpenAI, ChatOllama #model server 
+from langchain_groq import ChatGroq
+from langchain.chains import LLMChain
+from langchain.prompts import (
+    PromptTemplate,
+    SystemMessagePromptTemplate,
+    HumanMessagePromptTemplate,
+    ChatPromptTemplate,
+)
+from config import app_config
+import mongo_utils as mongo
+GROQ_API_KEY = "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"
+
+def __image2text(image):
+    """Generates a short description of the image"""
+    headers = {"Authorization": app_config.HF_TOKEN}
+    try:
+        response = requests.post(app_config.I2T_API_URL, headers=headers, data=image)
+        response = response.json()[0]["generated_text"]
+    except Exception as e:
+        print(e)
+    return response
+
+
+def __text2story(image_desc, genre, style, word_count, creativity):
+    """ "Generates a short story based on image description text prompt"""
+    ## chat LLM model
+    # story_model = ChatOpenAI(
+    #     model="gpt-3.5-turbo",
+    #     openai_api_key=app_config.OPENAI_KEY,
+    #     temperature=creativity,
+    # )
+
+    story_model = ChatGroq(model="llama3-8b-8192",  
+                temperature=0.0,
+                api_key=GROQ_API_KEY)
+    
+    ## chat message prompts
+    sys_prompt = PromptTemplate(
+        template="""You are an expert story writer, write a maximum of {word_count} 
+        words long story in {genre} genre in {style} writing style, based on the user 
+        provided story-context.
+        """,
+        input_variables=["word_count", "genre", "style"],
+    )
+    system_msg_prompt = SystemMessagePromptTemplate(prompt=sys_prompt)
+    human_prompt = PromptTemplate(
+        template="story-context: {context}", input_variables=["context"]
+    )
+    human_msg_prompt = HumanMessagePromptTemplate(prompt=human_prompt)
+    chat_prompt = ChatPromptTemplate.from_messages(
+        [system_msg_prompt, human_msg_prompt]
+    )
+    ## LLM chain
+    story_chain = LLMChain(llm=story_model, prompt=chat_prompt)
+    response = story_chain.run(
+        genre=genre, style=style, word_count=word_count, context=image_desc
+    )
+    return response
+
+
+def generate_story(image_file, genre, style, word_count, creativity):
+    """Generates a story given an image"""
+    # read image as bytes arrayS
+    with open(image_file, "rb") as f:
+        input_image = f.read()
+    # generate caption for image
+    image_desc = __image2text(image=input_image)
+
+    print("++++++++++++++++++++++++++++++++++++++")
+    print(image_desc)
+    print("++++++++++++++++++++++++++++++++++++++")
+    # generate story from caption
+    story = __text2story(
+        image_desc=image_desc,
+        genre=genre,
+        style=style,
+        word_count=word_count,
+        creativity=creativity,
+    )
+    # increment the openai access counter and compute count stats
+    mongo.increment_curr_access_count()
+    max_count = app_config.openai_max_access_count
+    curr_count = app_config.openai_curr_access_count
+    available_count = max_count - curr_count
+    return story, max_count, curr_count, available_count

mongo_utils.py

@@ -0,0 +1,57 @@
+import pymongo
+from config import app_config
+
+
+# Generic functions
+def get_db_client():
+    """Returns MongoDB client object, connect to MongoDB Atlas instances if required"""
+    try:
+        if app_config.mongo_client == None:
+            client = pymongo.MongoClient(app_config.MONGO_CONN_STR)
+            app_config.mongo_client = client
+    except Exception as e:
+        print(e)
+    return app_config.mongo_client
+
+
+def fetch_document(client, db, collection):
+    """Get a single document from the provided db and collection"""
+    try:
+        document = client[db][collection].find_one()
+    except Exception as e:
+        print(e)
+    return document
+
+
+def update_document(client, db, collection, key, value):
+    """Update the passed key in the document for provided db and collection"""
+    try:
+        document = fetch_document(client, db, collection)
+        client[db][collection].update_one(
+            {"_id": document["_id"]},
+            {"$set": {key: value}},
+        )
+    except Exception as e:
+        print(e)
+
+
+# Use case specific functions
+def fetch_curr_access_count():
+    client = get_db_client()
+    curr_count = fetch_document(
+        client=client, db=app_config.db, collection=app_config.collection
+    )[app_config.key]
+    app_config.openai_curr_access_count = curr_count
+
+
+def increment_curr_access_count():
+    client = get_db_client()
+    updated_count = app_config.openai_curr_access_count + 1
+    update_document(
+        client=client,
+        db=app_config.db,
+        collection=app_config.collection,
+        key=app_config.key,
+        value=updated_count,
+    )
+    app_config.openai_curr_access_count = updated_count

myenv/bin/Activate.ps1

@@ -0,0 +1,247 @@
+<#
+.Synopsis
+Activate a Python virtual environment for the current PowerShell session.
+
+.Description
+Pushes the python executable for a virtual environment to the front of the
+$Env:PATH environment variable and sets the prompt to signify that you are
+in a Python virtual environment. Makes use of the command line switches as
+well as the `pyvenv.cfg` file values present in the virtual environment.
+
+.Parameter VenvDir
+Path to the directory that contains the virtual environment to activate. The
+default value for this is the parent of the directory that the Activate.ps1
+script is located within.
+
+.Parameter Prompt
+The prompt prefix to display when this virtual environment is activated. By
+default, this prompt is the name of the virtual environment folder (VenvDir)
+surrounded by parentheses and followed by a single space (ie. '(.venv) ').
+
+.Example
+Activate.ps1
+Activates the Python virtual environment that contains the Activate.ps1 script.
+
+.Example
+Activate.ps1 -Verbose
+Activates the Python virtual environment that contains the Activate.ps1 script,
+and shows extra information about the activation as it executes.
+
+.Example
+Activate.ps1 -VenvDir C:\Users\MyUser\Common\.venv
+Activates the Python virtual environment located in the specified location.
+
+.Example
+Activate.ps1 -Prompt "MyPython"
+Activates the Python virtual environment that contains the Activate.ps1 script,
+and prefixes the current prompt with the specified string (surrounded in
+parentheses) while the virtual environment is active.
+
+.Notes
+On Windows, it may be required to enable this Activate.ps1 script by setting the
+execution policy for the user. You can do this by issuing the following PowerShell
+command:
+
+PS C:\> Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope CurrentUser
+
+For more information on Execution Policies: 
+https://go.microsoft.com/fwlink/?LinkID=135170
+
+#>
+Param(
+    [Parameter(Mandatory = $false)]
+    [String]
+    $VenvDir,
+    [Parameter(Mandatory = $false)]
+    [String]
+    $Prompt
+)
+
+<# Function declarations --------------------------------------------------- #>
+
+<#
+.Synopsis
+Remove all shell session elements added by the Activate script, including the
+addition of the virtual environment's Python executable from the beginning of
+the PATH variable.
+
+.Parameter NonDestructive
+If present, do not remove this function from the global namespace for the
+session.
+
+#>
+function global:deactivate ([switch]$NonDestructive) {
+    # Revert to original values
+
+    # The prior prompt:
+    if (Test-Path -Path Function:_OLD_VIRTUAL_PROMPT) {
+        Copy-Item -Path Function:_OLD_VIRTUAL_PROMPT -Destination Function:prompt
+        Remove-Item -Path Function:_OLD_VIRTUAL_PROMPT
+    }
+
+    # The prior PYTHONHOME:
+    if (Test-Path -Path Env:_OLD_VIRTUAL_PYTHONHOME) {
+        Copy-Item -Path Env:_OLD_VIRTUAL_PYTHONHOME -Destination Env:PYTHONHOME
+        Remove-Item -Path Env:_OLD_VIRTUAL_PYTHONHOME
+    }
+
+    # The prior PATH:
+    if (Test-Path -Path Env:_OLD_VIRTUAL_PATH) {
+        Copy-Item -Path Env:_OLD_VIRTUAL_PATH -Destination Env:PATH
+        Remove-Item -Path Env:_OLD_VIRTUAL_PATH
+    }
+
+    # Just remove the VIRTUAL_ENV altogether:
+    if (Test-Path -Path Env:VIRTUAL_ENV) {
+        Remove-Item -Path env:VIRTUAL_ENV
+    }
+
+    # Just remove VIRTUAL_ENV_PROMPT altogether.
+    if (Test-Path -Path Env:VIRTUAL_ENV_PROMPT) {
+        Remove-Item -Path env:VIRTUAL_ENV_PROMPT
+    }
+
+    # Just remove the _PYTHON_VENV_PROMPT_PREFIX altogether:
+    if (Get-Variable -Name "_PYTHON_VENV_PROMPT_PREFIX" -ErrorAction SilentlyContinue) {
+        Remove-Variable -Name _PYTHON_VENV_PROMPT_PREFIX -Scope Global -Force
+    }
+
+    # Leave deactivate function in the global namespace if requested:
+    if (-not $NonDestructive) {
+        Remove-Item -Path function:deactivate
+    }
+}
+
+<#
+.Description
+Get-PyVenvConfig parses the values from the pyvenv.cfg file located in the
+given folder, and returns them in a map.
+
+For each line in the pyvenv.cfg file, if that line can be parsed into exactly
+two strings separated by `=` (with any amount of whitespace surrounding the =)
+then it is considered a `key = value` line. The left hand string is the key,
+the right hand is the value.
+
+If the value starts with a `'` or a `"` then the first and last character is
+stripped from the value before being captured.
+
+.Parameter ConfigDir
+Path to the directory that contains the `pyvenv.cfg` file.
+#>
+function Get-PyVenvConfig(
+    [String]
+    $ConfigDir
+) {
+    Write-Verbose "Given ConfigDir=$ConfigDir, obtain values in pyvenv.cfg"
+
+    # Ensure the file exists, and issue a warning if it doesn't (but still allow the function to continue).
+    $pyvenvConfigPath = Join-Path -Resolve -Path $ConfigDir -ChildPath 'pyvenv.cfg' -ErrorAction Continue
+
+    # An empty map will be returned if no config file is found.
+    $pyvenvConfig = @{ }
+
+    if ($pyvenvConfigPath) {
+
+        Write-Verbose "File exists, parse `key = value` lines"
+        $pyvenvConfigContent = Get-Content -Path $pyvenvConfigPath
+
+        $pyvenvConfigContent | ForEach-Object {
+            $keyval = $PSItem -split "\s*=\s*", 2
+            if ($keyval[0] -and $keyval[1]) {
+                $val = $keyval[1]
+
+                # Remove extraneous quotations around a string value.
+                if ("'""".Contains($val.Substring(0, 1))) {
+                    $val = $val.Substring(1, $val.Length - 2)
+                }
+
+                $pyvenvConfig[$keyval[0]] = $val
+                Write-Verbose "Adding Key: '$($keyval[0])'='$val'"
+            }
+        }
+    }
+    return $pyvenvConfig
+}
+
+
+<# Begin Activate script --------------------------------------------------- #>
+
+# Determine the containing directory of this script
+$VenvExecPath = Split-Path -Parent $MyInvocation.MyCommand.Definition
+$VenvExecDir = Get-Item -Path $VenvExecPath
+
+Write-Verbose "Activation script is located in path: '$VenvExecPath'"
+Write-Verbose "VenvExecDir Fullname: '$($VenvExecDir.FullName)"
+Write-Verbose "VenvExecDir Name: '$($VenvExecDir.Name)"
+
+# Set values required in priority: CmdLine, ConfigFile, Default
+# First, get the location of the virtual environment, it might not be
+# VenvExecDir if specified on the command line.
+if ($VenvDir) {
+    Write-Verbose "VenvDir given as parameter, using '$VenvDir' to determine values"
+}
+else {
+    Write-Verbose "VenvDir not given as a parameter, using parent directory name as VenvDir."
+    $VenvDir = $VenvExecDir.Parent.FullName.TrimEnd("\\/")
+    Write-Verbose "VenvDir=$VenvDir"
+}
+
+# Next, read the `pyvenv.cfg` file to determine any required value such
+# as `prompt`.
+$pyvenvCfg = Get-PyVenvConfig -ConfigDir $VenvDir
+
+# Next, set the prompt from the command line, or the config file, or
+# just use the name of the virtual environment folder.
+if ($Prompt) {
+    Write-Verbose "Prompt specified as argument, using '$Prompt'"
+}
+else {
+    Write-Verbose "Prompt not specified as argument to script, checking pyvenv.cfg value"
+    if ($pyvenvCfg -and $pyvenvCfg['prompt']) {
+        Write-Verbose "  Setting based on value in pyvenv.cfg='$($pyvenvCfg['prompt'])'"
+        $Prompt = $pyvenvCfg['prompt'];
+    }
+    else {
+        Write-Verbose "  Setting prompt based on parent's directory's name. (Is the directory name passed to venv module when creating the virtual environment)"
+        Write-Verbose "  Got leaf-name of $VenvDir='$(Split-Path -Path $venvDir -Leaf)'"
+        $Prompt = Split-Path -Path $venvDir -Leaf
+    }
+}
+
+Write-Verbose "Prompt = '$Prompt'"
+Write-Verbose "VenvDir='$VenvDir'"
+
+# Deactivate any currently active virtual environment, but leave the
+# deactivate function in place.
+deactivate -nondestructive
+
+# Now set the environment variable VIRTUAL_ENV, used by many tools to determine
+# that there is an activated venv.
+$env:VIRTUAL_ENV = $VenvDir
+
+if (-not $Env:VIRTUAL_ENV_DISABLE_PROMPT) {
+
+    Write-Verbose "Setting prompt to '$Prompt'"
+
+    # Set the prompt to include the env name
+    # Make sure _OLD_VIRTUAL_PROMPT is global
+    function global:_OLD_VIRTUAL_PROMPT { "" }
+    Copy-Item -Path function:prompt -Destination function:_OLD_VIRTUAL_PROMPT
+    New-Variable -Name _PYTHON_VENV_PROMPT_PREFIX -Description "Python virtual environment prompt prefix" -Scope Global -Option ReadOnly -Visibility Public -Value $Prompt
+
+    function global:prompt {
+        Write-Host -NoNewline -ForegroundColor Green "($_PYTHON_VENV_PROMPT_PREFIX) "
+        _OLD_VIRTUAL_PROMPT
+    }
+    $env:VIRTUAL_ENV_PROMPT = $Prompt
+}
+
+# Clear PYTHONHOME
+if (Test-Path -Path Env:PYTHONHOME) {
+    Copy-Item -Path Env:PYTHONHOME -Destination Env:_OLD_VIRTUAL_PYTHONHOME
+    Remove-Item -Path Env:PYTHONHOME
+}
+
+# Add the venv to the PATH
+Copy-Item -Path Env:PATH -Destination Env:_OLD_VIRTUAL_PATH
+$Env:PATH = "$VenvExecDir$([System.IO.Path]::PathSeparator)$Env:PATH"

myenv/bin/activate

@@ -0,0 +1,63 @@
+# This file must be used with "source bin/activate" *from bash*
+# you cannot run it directly
+
+deactivate () {
+    # reset old environment variables
+    if [ -n "${_OLD_VIRTUAL_PATH:-}" ] ; then
+        PATH="${_OLD_VIRTUAL_PATH:-}"
+        export PATH
+        unset _OLD_VIRTUAL_PATH
+    fi
+    if [ -n "${_OLD_VIRTUAL_PYTHONHOME:-}" ] ; then
+        PYTHONHOME="${_OLD_VIRTUAL_PYTHONHOME:-}"
+        export PYTHONHOME
+        unset _OLD_VIRTUAL_PYTHONHOME
+    fi
+
+    # Call hash to forget past commands. Without forgetting
+    # past commands the $PATH changes we made may not be respected
+    hash -r 2> /dev/null
+
+    if [ -n "${_OLD_VIRTUAL_PS1:-}" ] ; then
+        PS1="${_OLD_VIRTUAL_PS1:-}"
+        export PS1
+        unset _OLD_VIRTUAL_PS1
+    fi
+
+    unset VIRTUAL_ENV
+    unset VIRTUAL_ENV_PROMPT
+    if [ ! "${1:-}" = "nondestructive" ] ; then
+    # Self destruct!
+        unset -f deactivate
+    fi
+}
+
+# unset irrelevant variables
+deactivate nondestructive
+
+VIRTUAL_ENV=$(cygpath "C:\Users\itsja\Documents\AI_IMG\pic-to-story\myenv")
+export VIRTUAL_ENV
+
+_OLD_VIRTUAL_PATH="$PATH"
+PATH="$VIRTUAL_ENV/bin:$PATH"
+export PATH
+
+# unset PYTHONHOME if set
+# this will fail if PYTHONHOME is set to the empty string (which is bad anyway)
+# could use `if (set -u; : $PYTHONHOME) ;` in bash
+if [ -n "${PYTHONHOME:-}" ] ; then
+    _OLD_VIRTUAL_PYTHONHOME="${PYTHONHOME:-}"
+    unset PYTHONHOME
+fi
+
+if [ -z "${VIRTUAL_ENV_DISABLE_PROMPT:-}" ] ; then
+    _OLD_VIRTUAL_PS1="${PS1:-}"
+    PS1="(myenv) ${PS1:-}"
+    export PS1
+    VIRTUAL_ENV_PROMPT="(myenv) "
+    export VIRTUAL_ENV_PROMPT
+fi
+
+# Call hash to forget past commands. Without forgetting
+# past commands the $PATH changes we made may not be respected
+hash -r 2> /dev/null

myenv/pyvenv.cfg

@@ -0,0 +1,5 @@
+home = C:\msys64\mingw64\bin
+include-system-site-packages = false
+version = 3.11.9
+executable = C:\msys64\mingw64\bin\python.exe
+command = C:\msys64\mingw64\bin\python.exe -m venv C:\Users\itsja\Documents\AI_IMG\pic-to-story\myenv

requirements.txt

@@ -0,0 +1,5 @@
+gradio==3.45.*
+requests==2.29.*
+langchain==0.0.*
+openai==0.27.*
+pymongo[tls, srv]==4.4.*

style.css

@@ -0,0 +1,11 @@
+.orange-button {
+  background-color: rgb(21, 255, 0) !important;
+}
+
+.gray-button {
+  background-color: #411d5a !important;
+}
+
+footer {
+  visibility: hidden;
+}

test.py

@@ -0,0 +1,13 @@
+import openai
+
+openai.api_key = 'fake-api'
+openai.base_url = "http://localhost:3040/v1/"
+
+completion = openai.chat.completions.create(
+    model="gpt-3.5-turbo",
+    messages=[
+        {"role": "user", "content": "How do I list all files in a directory using Python?"},
+    ],
+)
+
+print(completion.choices[0].message.content)