Upload 3 files

app.py

@@ -0,0 +1,55 @@
+import streamlit as st
+import llm
+from file_creator import Create_Doc
+
+model = llm.Model()
+unvalid_image_text = 'please upload a valid screenshot.'
+
+
+st.markdown("""
+ <style>
+ .justified-text {
+ text-align: justify;
+ }
+ </style>
+ """, unsafe_allow_html=True)
+
+
+with st.sidebar:
+ st.header("ABOUT:")
+ 
+ st.caption("""
+ <div class="justified-text">
+ Document Creator is an innovative app that allows users to effortlessly convert their screenshots into Word documents. Simply upload a screenshot, and the app will generate a Word document based on the image provided, ensuring a seamless and efficient conversion process. Ideal for anyone looking to quickly turn visual content into editable text documents.
+ </div>
+ """, unsafe_allow_html=True)
+ 
+ for _ in range(17):
+ st.write("") 
+ st.subheader("Build By:")
+ st.write("[Pachaiappan❤️](https://mr-vicky-01.github.io/Portfolio)")
+ st.write("contact: [Email](mailto:[email protected])")
+
+st.title("Document Creator")
+st.text("Updload your Screenshot to convert into a word document.")
+uploaded_file = st.file_uploader("", type=["png", "jpg", "jpeg"])
+
+if uploaded_file:
+ st.image(uploaded_file)
+ 
+ button = st.button("Generate Document")
+ if button:
+ with st.spinner("Generating a Document"):
+ text = model.get_response(uploaded_file)
+ st.write(text)
+ 
+ if text.lower().strip() != unvalid_image_text:
+ doc = Create_Doc()
+ doc_buffer = doc.markdown_to_word(text)
+ st.download_button(
+ label="Download",
+ data=doc_buffer,
+ file_name="output.docx",
+ mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+ )
+ 

file_creator.py

@@ -0,0 +1,40 @@
+from docx import Document
+from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
+from docx.shared import Pt
+from io import BytesIO
+import markdown
+
+class Create_Doc:
+ def __init__(self) -> None:
+ self.doc = Document()
+ 
+ def markdown_to_word(self,markdown_text):
+ # Convert Markdown to HTML
+ # html = markdown.markdown(markdown_text)
+
+ # Parse the Markdown text and add formatted content to the document
+ for line in markdown_text.split('\n'):
+ if line.startswith('# '):
+ heading = line[2:]
+ p = self.doc.add_heading(heading, level=1)
+ elif line.startswith('## '):
+ heading = line[3:]
+ p = self.doc.add_heading(heading, level=2)
+ elif line.startswith('### '):
+ heading = line[4:]
+ p = self.doc.add_heading(heading, level=3)
+ elif line.startswith('- '):
+ item = line[2:]
+ p = self.doc.add_paragraph(item, style='ListBullet')
+ else:
+ p = self.doc.add_paragraph(line)
+ 
+ # Adjust paragraph formatting (optional)
+ p.alignment = WD_PARAGRAPH_ALIGNMENT.LEFT
+ p.style.font.size = Pt(12)
+
+ # Save the document to a BytesIO object
+ buffer = BytesIO()
+ self.doc.save(buffer)
+ buffer.seek(0)
+ return buffer

llm.py

@@ -0,0 +1,19 @@
+import google.generativeai as genai
+from dotenv import load_dotenv
+from PIL import Image
+import os
+
+load_dotenv()
+os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
+genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+
+class Model:
+ def __init__(self) -> None:
+ self.model = genai.GenerativeModel('gemini-1.5-flash-latest')
+ 
+ def get_response(self, image):
+ prompt = """You are an intelligent document creator. Could you please extract the words from the given screenshot and provide me document text that matches exact screenshot font and look
+ important note: if the screenshot not contain any text means you must say 'please upload a valid screenshot.'"""
+ img = Image.open(image)
+ response = self.model.generate_content([prompt, img]) 
+ return response.text