Upload 5 files

app.py

@@ -0,0 +1,112 @@
+import streamlit as st
+from model import AllModels
+from html_pdf import Pdf
+from pdf_process import PdfExtract
+from PIL import Image
+model = AllModels()
+roadmap_generator = Pdf()
+
+
+st.title('NoteFusion')
+st.markdown("""
+    <style>
+    .justified-text {
+        text-align: justify;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+# Sidebar Section
+st.sidebar.title("About")
+
+st.sidebar.caption("""
+        <div class="justified-text">
+           NoteFusion is an AI-powered educational assistant helps students by providing summarizing textbook content, and generating personalized study plans roadmap Pdf and converts image to Notes. Powered by the Gemini API and built with Streamlit.
+        </div>
+        """, unsafe_allow_html=True)
+
+for _ in range(3):
+        st.sidebar.write("") 
+
+menu = ["Image To Notes", "Summarize Textbook", "Generate Roadmap PDF","Generate own book"]
+choice = st.sidebar.selectbox("Choose an option", menu)
+
+for _ in range(3):
+        st.sidebar.write("") 
+
+if choice == 'Generate Roadmap PDF':
+    st.subheader('Generate Roadmap PDF')
+    subjects = st.text_input("Enter the subjects you need help with (comma-separated)")
+    if subjects:
+        button = st.button("Generate RoadMap")
+    if subjects and button:
+        with st.spinner('creating...'):
+            response = model.road_map_model(subjects)
+            print(response)
+            response = response.replace("```html", "")
+            response = response.replace("```", "")
+            pdf_bytes = roadmap_generator.pdf_creator(response)
+            st.success("PDF generated successfully!")
+
+            st.download_button("Download PDF", data=pdf_bytes, file_name="output.pdf", mime="application/pdf")
+
+elif choice == "Summarize Textbook":
+    st.subheader("Summarize Textbook")
+    uploaded_file = st.file_uploader("Upload a textbook pdf", type=["pdf",'png','jpg','jpeg'])
+    if uploaded_file:
+        button = st.button("Generate Summary")
+        type = uploaded_file.name.split('.')[-1]
+        if type!='pdf':
+            st.image(uploaded_file,caption='uploaded image')
+        else:
+             pdf_text = PdfExtract(uploaded_file)
+    if uploaded_file and button:
+        with st.spinner('creating...'):
+            if type == 'pdf':
+                response = model.summary_model(input_doc = pdf_text.extracted_text)
+                st.write(response)
+            else:
+                img = Image.open(uploaded_file)
+                response = model.summary_model(image=img)
+                st.write(response)
+                 
+
+elif choice == "Generate own book":
+    st.subheader("Create your own book")
+    st.caption('Give the book name and pages our AI will create book for you')
+    book_name = st.text_input("textbook Naame")
+    if book_name:
+        button = st.button("Generate Book")
+    if book_name and button:
+        with st.spinner('creating...'):
+            response = model.book_creator(book_name)
+            st.write(response)
+            response = response.replace("```html", "")
+            response = response.replace("```", "")
+           
+            pdf_bytes = roadmap_generator.pdf_creator(response)
+            st.success("PDF generated successfully!")
+
+            st.download_button("Download PDF", data=pdf_bytes, file_name="output.pdf", mime="application/pdf")
+
+elif choice == "Image To Notes":
+    st.subheader("Image To Notes")
+    uploaded_file = st.file_uploader("Upload a image or screenshot", type=['png','jpg','jpeg'])
+    if uploaded_file:
+        st.image(uploaded_file,caption='uploaded image')
+        
+        button = st.button("Generate Notes")
+
+    if uploaded_file and button:
+        with st.spinner('creating...'):
+            img = Image.open(uploaded_file)
+            response = model.note_model(img)
+            st.write(response)
+
+        
+
+     
+
+   
+
+

html_pdf.py

@@ -0,0 +1,50 @@
+from fpdf import FPDF
+from fpdf.html import HTMLMixin
+from bs4 import BeautifulSoup
+from io import BytesIO
+
+
+class MyFPDF(FPDF, HTMLMixin):
+    def header(self):
+        # Add a border around the page
+        self.rect(5.0, 5.0, self.w - 10.0, self.h - 10.0)
+
+class Pdf:
+    def __init__(self):
+        self.pdf = MyFPDF()
+
+    def replace_special_characters(self, text):
+        replacements = {
+            '’': "'",
+            '“': '"',
+            '”': '"',
+            '–': '-',
+            '—': '-',
+            '…': '...',
+        }
+        for original, replacement in replacements.items():
+            text = text.replace(original, replacement)
+        return text
+
+    def pdf_creator(self, html_text):
+        self.pdf.add_page()
+    
+        # Use the built-in Arial font
+        self.pdf.set_font("Times", size=12)
+       
+        soup = BeautifulSoup(html_text, 'html.parser')
+
+          # Replace special characters in the text
+        for element in soup.find_all(text=True):
+            element.replace_with(self.replace_special_characters(element))
+
+        # Convert entire HTML content to PDF
+        self.pdf.write_html(str(soup))
+
+        # Save PDF to bytes object
+        output = BytesIO()
+        self.pdf.output(output)
+        pdf_bytes = output.getvalue()
+        output.close()
+
+        return pdf_bytes

model.py

@@ -0,0 +1,73 @@
+import google.generativeai as genai
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
+genai.configure(api_key="AIzaSyDA6zE16NjMSYefmnIo74aJ8cacoDxJsrE")
+
+
+class AllModels:
+    def __init__(self):
+       self.model = genai.GenerativeModel('gemini-1.5-flash-latest')
+
+
+    def road_map_model(self,subjects):
+        prompt = f"""Create an HTML format roadmap for learning {subjects} with proving week task . The roadmap should include the following sections:
+
+        Introduction: Brief overview of the subject and its importance.
+
+        Foundation: Fundamental concepts and theories to grasp before diving deeper.
+
+        Intermediate Level: More advanced topics building upon the foundation.
+
+        Advanced Level: Complex concepts, techniques, or applications within the subject.
+
+        Resources: List of recommended books, online courses, tutorials, and websites for further learning using hyperlink tag.
+
+        Projects: Hands-on projects or exercises to reinforce learning and apply knowledge.
+
+        Conclusion: Summary of key takeaways and encouragement for further exploration.
+
+        The HTML format should be visually appealing with appropriate headings, subheadings, bullet points, and hyperlinks where necessary. Ensure clarity and logical progression in the roadmap.
+        NOTE: The roadmap should be provided in pure HTML format without any Markdown or other formatting styles and In the html format dont need style tags and title tags"""
+
+        response = self.model.generate_content([prompt.format(subjects=subjects)]) 
+        return response.text
+    
+
+    def summary_model(self,input_doc=None,image=None):
+        if input_doc:
+            prompt = """you are an intelligent assistant using the giving document summarize it shortly
+            document:
+            {document}
+            """
+            response = self.model.generate_content([prompt.format(document=input_doc)]) 
+            return response.text
+        elif image:
+             prompt = """you are an intelligent assistant using the giving image summarize it and its content."""
+             response = self.model.generate_content([prompt,image]) 
+             return response.text
+
+
+    
+
+    def book_creator(self,book_name):
+        prompt = """"Generate a detailed and extensive book titled '{book_title}'.
+        Please provide the content in HTML format with appropriate tags for headings, paragraphs,
+        NOTE: The book should be provided in pure "HTML" format without any "Markdown" or other formatting styles and In the html format dont need style tags and title tags
+        Ensure the HTML is well-structured and valid. Here is the beginning of the book in HTML:\n\n
+        <html>\n<head>\n<title>'{book_title}'</title>\n</head>\n<body>\n"
+        <h1>'{book_title}'</h1>\n"
+        <div>\n"""
+
+        response = self.model.generate_content([prompt.format(book_title=book_name)]) 
+        return response.text
+    
+    def note_model(self,img):
+        prompt = """You are an intelligent Notes creator.create a notes using given screenshot
+        important note: if the screenshot not contain any text means you must say 'please upload a valid screenshot."""
+
+        response = self.model.generate_content([prompt,img]) 
+        return response.text
+

pdf_process.py

@@ -0,0 +1,15 @@
+from PyPDF2 import PdfReader
+
+class PdfExtract:
+
+    def __init__(self, pdf_file):
+        self.extracted_text = self.get_pdf_text(pdf_file)
+
+    def get_pdf_text(self, pdf_file):
+        text = ""
+        pdf_reader = PdfReader(pdf_file)
+        for page in pdf_reader.pages:
+            page_text = page.extract_text()
+            if page_text:
+                text += page_text
+        return text

requirements.txt

@@ -0,0 +1,6 @@
+streamlit
+google-generativeai
+beautifulsoup4
+python-dotenv
+fpdf
+PyPDF2