Update app.py

app.py

@@ -1,106 +1,104 @@
-import streamlit as st
-import google.generativeai as genai
-from dotenv import load_dotenv
-from PIL import Image
-import markdown
-from docx import Document
-from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
-from docx.shared import Pt
-from io import BytesIO
-import os
-
-load_dotenv()
-os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
-
-genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
-
-model = genai.GenerativeModel('gemini-1.5-flash-latest')
-
-def response(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 = model.generate_content([prompt, img]) 
-    return response.text
-
-# Function to convert Markdown to Word document
-def markdown_to_word(markdown_text):
-    # Create a new Word document
-    doc = Document()
-
-    for line in markdown_text.split('\n'):
-        if line.startswith('# '):
-            heading = line[2:]
-            p = doc.add_heading(heading, level=1)
-        elif line.startswith('## '):
-            heading = line[3:]
-            p = doc.add_heading(heading, level=2)
-        elif line.startswith('### '):
-            heading = line[4:]
-            p = doc.add_heading(heading, level=3)
-        elif line.startswith('- '):
-            item = line[2:]
-            p = doc.add_paragraph(item, style='ListBullet')
-        else:
-            p = doc.add_paragraph()
-            words = line.split(' ')
-            for word in words:
-                if word.startswith('**') and word.endswith('**'):
-                    run = p.add_run(word[2:-2])
-                    run.bold = True
-                elif word.startswith('*') and word.endswith('*'):
-                    run = p.add_run(word[1:-1])
-                    run.italic = True
-                else:
-                    p.add_run(word)
-                p.add_run(' ')
-
-    # Save the document to a BytesIO object
-    buffer = BytesIO()
-    doc.save(buffer)
-    buffer.seek(0)
-    return buffer
-
-st.title("Document Creator")
-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:pachaiappan1102@gamil.com)")
-    
-fake_image_text = 'please upload a valid screenshot.'
-    
-uploaded_file = st.file_uploader("Updload your Screenshot", 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 = response(uploaded_file)
-        st.write(text)
-        
-        if text.lower().strip() != fake_image_text:
-            doc_buffer = markdown_to_word(text)
-            st.download_button(
-            label="Download",
-            data=doc_buffer,
-            file_name="output.docx",
-            mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-            )
+import streamlit as st
+import google.generativeai as genai
+from PIL import Image
+import markdown
+from docx import Document
+from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
+from docx.shared import Pt
+from io import BytesIO
+import os
+
+os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
+
+genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+
+model = genai.GenerativeModel('gemini-1.5-flash-latest')
+
+def response(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 = model.generate_content([prompt, img]) 
+    return response.text
+
+# Function to convert Markdown to Word document
+def markdown_to_word(markdown_text):
+    # Create a new Word document
+    doc = Document()
+
+    for line in markdown_text.split('\n'):
+        if line.startswith('# '):
+            heading = line[2:]
+            p = doc.add_heading(heading, level=1)
+        elif line.startswith('## '):
+            heading = line[3:]
+            p = doc.add_heading(heading, level=2)
+        elif line.startswith('### '):
+            heading = line[4:]
+            p = doc.add_heading(heading, level=3)
+        elif line.startswith('- '):
+            item = line[2:]
+            p = doc.add_paragraph(item, style='ListBullet')
+        else:
+            p = doc.add_paragraph()
+            words = line.split(' ')
+            for word in words:
+                if word.startswith('**') and word.endswith('**'):
+                    run = p.add_run(word[2:-2])
+                    run.bold = True
+                elif word.startswith('*') and word.endswith('*'):
+                    run = p.add_run(word[1:-1])
+                    run.italic = True
+                else:
+                    p.add_run(word)
+                p.add_run(' ')
+
+    # Save the document to a BytesIO object
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return buffer
+
+st.title("Document Creator")
+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])")
+    
+fake_image_text = 'please upload a valid screenshot.'
+    
+uploaded_file = st.file_uploader("Updload your Screenshot", 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 = response(uploaded_file)
+        st.write(text)
+        
+        if text.lower().strip() != fake_image_text:
+            doc_buffer = markdown_to_word(text)
+            st.download_button(
+            label="Download",
+            data=doc_buffer,
+            file_name="output.docx",
+            mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+            )