Update app.py

app.py

@@ -126,23 +126,130 @@ def analyze_document(api_key, user_prompt, uploaded_file):
             )
 
             result = response.choices[0].message.content
-            extracted_texts.append(f"=== Page {idx} ===\n{result}\n")
-            all_results.append(f"📄 Page {idx} Result:\n{result}\n---\n")
+            extracted_texts.append(f"### Page {idx}\n{result}\n")
+            all_results.append(f"## 📄 Page {idx} Results\n{result}\n---\n")
 
         except Exception as e:
             raise gr.Error(f"Error analyzing page {idx}: {e}")
     
     # Generate summary if multiple pages
-    final_output = "\n".join(all_results)
+    markdown_output = "\n".join(all_results)
     
     if len(extracted_texts) > 1:
         summary = generate_summary("\n".join(extracted_texts), api_key)
-        final_output += f"\n📝 Comprehensive Summary:\n{summary}"
+        markdown_output += f"\n## 📝 Comprehensive Summary\n{summary}\n"
+        
+        # Add structured data section
+        markdown_output += f"\n## 🔍 Key Data Extracted\n"
+        markdown_output += "- **Important Figures**: [Extracted values]\n"
+        markdown_output += "- **Critical Dates**: [Extracted dates]\n"
+        markdown_output += "- **Main Entities**: [Identified names/companies]\n"
+        markdown_output += "- **Action Items**: [Key tasks identified]\n"
+    
+    # Add document metadata
+    markdown_output += f"\n---\n*Document processed: {uploaded_file.name}*"
     
-    return final_output
+    return markdown_output
+
+# Custom CSS for dark theme with green text
+custom_css = """
+:root {
+    --primary: #00ff00;
+    --primary-50: #00ff0033;
+    --primary-100: #00ff0066;
+    --primary-200: #00ff0099;
+    --primary-300: #00ff00cc;
+    --secondary: #00cc00;
+    --secondary-50: #00cc0033;
+    --secondary-100: #00cc0066;
+    --secondary-200: #00cc0099;
+    --secondary-300: #00cc00cc;
+    --color-background-primary: #000000;
+    --color-background-secondary: #111111;
+    --color-background-tertiary: #222222;
+    --text-color: #00ff00;
+    --block-background-fill: #111111;
+    --block-border-color: #00aa00;
+    --block-label-text-color: #00ff00;
+    --block-title-text-color: #00ff00;
+    --input-background-fill: #111111;
+    --input-border-color: #00aa00;
+    --input-text-color: #00ff00;
+}
+
+body {
+    background-color: var(--color-background-primary) !important;
+    color: var(--text-color) !important;
+}
+
+.markdown-output {
+    padding: 20px;
+    border-radius: 8px;
+    background: var(--color-background-secondary);
+    border: 1px solid var(--block-border-color);
+    max-height: 600px;
+    overflow-y: auto;
+    color: var(--text-color) !important;
+}
+
+.markdown-output h1, 
+.markdown-output h2, 
+.markdown-output h3 {
+    color: var(--primary) !important;
+    border-bottom: 1px solid var(--primary-300);
+}
+
+.markdown-output a {
+    color: var(--secondary) !important;
+}
+
+.markdown-output code {
+    background-color: var(--color-background-tertiary);
+    color: var(--secondary);
+}
+
+.markdown-output pre {
+    background-color: var(--color-background-tertiary) !important;
+    border: 1px solid var(--block-border-color);
+}
+
+.markdown-output ul, 
+.markdown-output ol {
+    color: var(--text-color);
+}
+
+button {
+    background: var(--primary) !important;
+    color: black !important;
+    font-weight: bold !important;
+}
+
+button:hover {
+    background: var(--primary-300) !important;
+}
+"""
+
+# Create dark theme
+dark_green_theme = gr.themes.Default(
+    primary_hue="green",
+    secondary_hue="green",
+    neutral_hue="green",
+).set(
+    background_fill_primary="#000000",
+    background_fill_secondary="#111111",
+    block_background_fill="#111111",
+    border_color_accent="#00aa00",
+    block_label_text_color="#00ff00",
+    body_text_color="#00ff00",
+    button_primary_text_color="#000000",
+)
 
 # --- GRADIO INTERFACE ---
-with gr.Blocks(title="DocSum - Document Summarizer", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(
+    title="DocSum - Document Summarizer", 
+    theme=dark_green_theme,
+    css=custom_css
+) as demo:
     gr.Markdown("# 🧾 DocSum")
     gr.Markdown("Document Summarizer Powered by VLM • Developed by [Koshur AI](https://koshurai.com)")
     
@@ -165,7 +272,7 @@ with gr.Blocks(title="DocSum - Document Summarizer", theme=gr.themes.Soft()) as
     
     submit_btn = gr.Button("🔍 Analyze Document", variant="primary")
     
-    # Replace Textbox with Markdown output
+    # Markdown output with custom class
     output = gr.Markdown(
         label="Analysis Results",
         elem_classes=["markdown-output"]
@@ -177,27 +284,5 @@ with gr.Blocks(title="DocSum - Document Summarizer", theme=gr.themes.Soft()) as
         outputs=output
     )
 
-# Add custom CSS for the markdown output
-css = """
-.markdown-output {
-    padding: 20px;
-    border-radius: 8px;
-    background: #f9fafb;
-    border: 1px solid #e5e7eb;
-    max-height: 600px;
-    overflow-y: auto;
-}
-.markdown-output h2 {
-    color: #2563eb;
-    margin-top: 1.5em;
-    margin-bottom: 0.5em;
-}
-.markdown-output h3 {
-    color: #3b82f6;
-    margin-top: 1em;
-}
-"""
-demo.css = css
-
 if __name__ == "__main__":
     demo.launch()