External files for Streamlit

.gitignore

@@ -0,0 +1,89 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# 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/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# DotEnv configuration
+.env
+
+# Database
+*.db
+*.rdb
+
+# Pycharm
+.idea
+
+# VS Code
+.vscode/
+
+# Spyder
+.spyproject/
+
+# Jupyter NB Checkpoints
+.ipynb_checkpoints/
+
+# exclude data from source control by default
+/data/
+
+# Mac OS-specific storage files
+.DS_Store
+
+# vim
+*.swp
+*.swo
+
+# Mypy cache
+.mypy_cache/

README.md

@@ -1,13 +0,0 @@
----
-title: TopicGen
-emoji: 🏆
-colorFrom: gray
-colorTo: gray
-sdk: streamlit
-sdk_version: 1.26.0
-app_file: app.py
-pinned: false
-license: mit
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

main.py

@@ -0,0 +1,47 @@
+from utils.image_utils import load_image, check_url
+from utils.caption_utils import ImageCaptioning
+from utils.topic_generation import TopicGenerator
+import streamlit as st
+
+
+def main():
+    st.title("TopicGen")
+    # Create an instance of the TopicGenerator class
+    topic_generator = TopicGenerator()
+    img_caption = ImageCaptioning()
+
+    # User input
+    user_input = st.selectbox(label="Text Input or Image Input", options=["Text", "Image", "Image URL"])
+    if user_input == "Text":
+        text_input = st.text_input(label="Put in your Idea, Let's generate a matching Topic Sentence🤗🤗")
+        generated_topics = topic_generator.generate_topics(text_input)
+        for idx, topic in enumerate(generated_topics, 1):
+            st.write(f"Topic {idx}: {topic}")
+    elif user_input == "Image":
+        img_input = st.file_uploader(label="Drop an Image you have been admiring, Let's see what we can do🤔🤔",
+                                     type=["jpg", "png", "jpeg"],
+                                     accept_multiple_files=True)
+        for in_img in img_input:
+            if in_img is not None:
+                img = load_image(in_img)
+                capt = img_caption.get_caption(img)
+                st.image(image=img, caption=capt, width=250, height=250)
+                generated_topics = topic_generator.generate_topics(capt)
+                for idx, topic in enumerate(generated_topics, 1):
+                    st.write(f"Topic {idx}: {topic}")
+
+    elif user_input == "Image URL":
+        url_input = st.text_input(label="Do you have a link to the Image you would like to drop, Go Ahead and We got "
+                                        "you covered😉😉")
+        url_img = check_url(url_input)
+        img_load = load_image(url_img)
+        caption = img_caption.get_caption(img_load)
+        st.image(image=img_load, caption=caption, width=250, height=250)
+        # Generate and display topics
+        generated_topics = topic_generator.generate_topics(caption)
+        for idx, topic in enumerate(generated_topics, 1):
+            st.write(f"Topic {idx}: {topic}")
+
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,5 @@
+requests~=2.27.1
+pillow~=9.0.1
+torch
+transformers
+streamlit