UPDATE: main.py

main.py

@@ -15,37 +15,71 @@ import time
 
 # Configure the page
 st.set_page_config(
-    page_title="PDF Summarizer",
-    page_icon=None,
+    page_title="AlphaExtract — Your AI-powered PDF Summarizer",
+    page_icon="📈",
     layout="wide"
 )
 
-# Initialize the pipeline
-pipeline = Pipeline()
-
 # Custom styling
 st.markdown("""
     <style>
+    html, body, [class*="css"]  {
+        font-family: 'Segoe UI', sans-serif;
+        background-color: #f9fbfd;
+    }
+
     .main-header {
-        font-size: 2.5rem;
-        color: #1E88E5;
-        margin-bottom: 2rem;
+        font-size: 3rem;
+        color: #0A66C2;
+        font-weight: 700;
+        text-align: center;
+        margin-top: 1rem;
+        margin-bottom: 2.5rem;
     }
+
     .summary-header {
         font-size: 1.8rem;
-        color: #2E7D32;
-        margin: 1.5rem 0;
+        color: #00695C;
+        font-weight: 600;
+        border-bottom: 2px solid #e0e0e0;
+        padding-bottom: 0.5rem;
+        margin-top: 2rem;
     }
+
     .status-container {
-        padding: 1rem;
-        border-radius: 0.5rem;
-        background-color: #f0f2f6;
-        margin: 1rem 0;
+        background: rgba(255, 255, 255, 0.7);
+        backdrop-filter: blur(10px);
+        border: 1px solid #E3EAF2;
+        box-shadow: 0px 4px 12px rgba(0,0,0,0.05);
+        border-radius: 10px;
+        padding: 1.2rem 1.5rem;
+        margin: 1.5rem 0;
+    }
+
+    .stDownloadButton > button {
+        background-color: #0A66C2;
+        color: white;
+        font-weight: bold;
+        border-radius: 8px;
+        padding: 0.6rem 1.2rem;
+        margin-top: 1rem;
+        transition: background-color 0.3s ease;
+    }
+
+    .stDownloadButton > button:hover {
+        background-color: #084B8A;
+    }
+
+    .sidebar .sidebar-content {
+        background-color: #ffffff;
     }
     </style>
 """, unsafe_allow_html=True)
 
-# Cache the font download
+# Initialize the pipeline
+pipeline = Pipeline()
+
+# Cache font download
 @st.cache_resource(show_spinner=False)
 def getDejaVuFontPath():
     fontUrl = "https://github.com/senotrusov/dejavu-fonts-ttf/raw/refs/heads/master/ttf/DejaVuSans.ttf"
@@ -55,12 +89,12 @@ def getDejaVuFontPath():
     tempFontFile.close()
     return tempFontFile.name
 
-# Cache the summary generation
+# Cache summary generation
 @st.cache_data(show_spinner=False, ttl=3600)
 def generateSummary(_pipeline, pdfBytes):
     return pipeline.run(pdfBytes)
 
-# Cache the PDF generation
+# Cache PDF generation
 @st.cache_data(show_spinner=False, ttl=3600)
 def generatePdfBytes(summary, fontPath):
     buffer = BytesIO()
@@ -92,65 +126,84 @@ def generatePdfBytes(summary, fontPath):
 
 # Sidebar
 with st.sidebar:
-    st.markdown("### Upload PDF")
-    uploadedFile = st.file_uploader("Choose your PDF file", type=["pdf"])
+    st.markdown("## 📄 Upload PDF")
+    uploadedFile = st.file_uploader("Drop your financial PDF here", type=["pdf"])
 
     if uploadedFile:
+        st.markdown("### 🔍 File Info")
         pdfDetails = {
-            "File Name": uploadedFile.name,
-            "File Size": f"{round(len(uploadedFile.getvalue()) / 1024, 2)} KB",
-            "Upload Time": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            "📁 File Name": uploadedFile.name,
+            "📦 Size": f"{round(len(uploadedFile.getvalue()) / 1024, 2)} KB",
+            "⏰ Uploaded": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         }
-
-        st.markdown("### PDF Details")
         for key, value in pdfDetails.items():
-            st.write(f"**{key}:** {value}")
+            st.write(f"**{key}**: {value}")
+
+    st.markdown("---")
+    st.markdown("#### ⚙️ Powered By")
+
+    col1, col2 = st.columns([1, 5])
+    with col1:
+        st.image("mnt/data/d0758679-eff1-46c8-9e0c-47d782259782.png", width=45)
+
+    with col2:
+        st.markdown("""
+        <div style='font-size: 0.9rem; line-height: 1.4;'>
+        Inference by <strong>Groq</strong><br>
+        using Meta’s <strong>LLaMA 4 MOE Maverick</strong><br>
+        for blazing-fast, high-precision summaries.
+        </div>
+        """, unsafe_allow_html=True)
+
 
 # Main content
-st.markdown("<h1 class='main-header'>PDF Summarizer</h1>", unsafe_allow_html=True)
-st.write("This application generates a concise summary from your uploaded PDF document. Upload a file to get started.")
+st.markdown("<h1 class='main-header'>Welcome to <span style='color:#2E7D32'>AlphaExtract</span></h1>", unsafe_allow_html=True)
+st.write("Upload a PDF to instantly receive a professional-grade analytical summary.")
 
 if uploadedFile:
     statusContainer = st.empty()
     summaryContainer = st.empty()
 
     with statusContainer.container():
-        st.markdown("### Processing Status")
+        st.markdown("<div class='status-container'>", unsafe_allow_html=True)
+        st.markdown("### ⏳ Processing Status")
         statusBox = st.empty()
 
         try:
             startTime = time.time()
-            statusBox.info("Reading PDF file...")
+            statusBox.info("📘 Reading PDF file...")
             pdfBytes = uploadedFile.getvalue()
             readDuration = time.time() - startTime
-            statusBox.success(f"PDF file read successfully ({readDuration:.2f}s)")
+            statusBox.success(f"✅ PDF file read successfully ({readDuration:.2f}s)")
 
-            statusBox.info("Generating summary...")
+            statusBox.info("🧠 Generating summary...")
             summary = generateSummary(pipeline, pdfBytes)
             totalTime = time.time() - startTime
 
             if summary:
-                statusBox.success(f"Summary generated successfully (Total time: {totalTime:.2f}s)")
+                statusBox.success(f"✅ Summary generated successfully (Total time: {totalTime:.2f}s)")
+                st.markdown("</div>", unsafe_allow_html=True)
 
                 with summaryContainer.container():
-                    st.markdown("<h2 class='summary-header'>Generated Summary</h2>", unsafe_allow_html=True)
+                    st.markdown("<h2 class='summary-header'>📊 Generated Summary</h2>", unsafe_allow_html=True)
                     st.markdown(summary)
+
                     try:
                         fontPath = getDejaVuFontPath()
                         pdfBytesOut = generatePdfBytes(summary, fontPath)
                         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
 
                         st.download_button(
-                            label="Download Summary as PDF",
+                            label="⬇️ Download Summary as PDF",
                             data=pdfBytesOut,
                             file_name=f"summary_{timestamp}.pdf",
                             mime="application/pdf"
                         )
                     except Exception as e:
-                        st.error(f"Error creating PDF: {str(e)}")
+                        st.error(f"❌ Error creating PDF: {str(e)}")
             else:
-                statusBox.error("Failed to generate summary. Please try again.")
+                statusBox.error("❌ Failed to generate summary. Please try again.")
         except Exception as e:
-            statusBox.error(f"Error processing PDF: {str(e)}")
+            statusBox.error(f"❌ Error processing PDF: {str(e)}")
 else:
-    st.info("Please upload a PDF file using the sidebar to get started.")
+    st.info("🚀 Please upload a PDF file using the sidebar to get started.")