Update app.py

app.py

@@ -1,138 +1,187 @@
 import streamlit as st
-import json
 import pandas as pd
 import numpy as np
 import io
-from omni_agent_v8 import OMNIOrchestratorV8
-from omni_genomics import OMNIGenomics
-
-# PDF and QR Libraries
+import qrcode
 from reportlab.lib.pagesizes import A4
 from reportlab.pdfgen import canvas
 from reportlab.lib.units import inch
-import qrcode
+from reportlab.lib.utils import ImageReader
+from omni_agent_v8 import OMNIOrchestratorV8
+from omni_genomics import OMNIGenomics
 
-# Page Configuration
-st.set_page_config(page_title="Abyssinia Intelligence V8", layout="wide")
+st.set_page_config(page_title="Abyssinia V8 | Precision", layout="wide")
 
-# Custom Gemini-style Response Styling
+# CSS: V8 "Stealth" UI + Gemini-Style Output Cards
 st.markdown("""
     <style>
-    .gemini-response {
-        background-color: #ffffff;
-        border: 1px solid #e3e3e3;
+    /* Global Stealth Theme */
+    .stApp { background-color: #050505; color: #e3e3e3; }
+    
+    /* Rounded Buttons */
+    .stButton button { 
+        border-radius: 30px; 
+        background: #1a1a1c; 
+        border: 1px solid #333; 
+        color: #e3e3e3;
+        transition: all 0.3s;
+    }
+    .stButton button:hover { border-color: #4285f4; color: #4285f4; }
+    .stButton button:active { background: #4285f4 !important; color: white; }
+
+    /* GEMINI-LIKE CARD (Dark Mode Optimized) */
+    .gemini-card {
+        background-color: #161618;
+        border: 1px solid #2d2d2d;
+        border-radius: 16px;
         padding: 25px;
-        border-radius: 12px;
-        color: #1f1f1f;
-        font-family: sans-serif;
+        margin-top: 10px;
+        box-shadow: 0 4px 20px rgba(0,0,0,0.5);
+        font-family: 'Segoe UI', Roboto, Helvetica, Arial, sans-serif;
         line-height: 1.6;
     }
-    .stButton>button { border-radius: 20px; background-color: #1a73e8; color: white; }
+    .gemini-card h1, .gemini-card h2, .gemini-card h3 {
+        color: #8ab4f8; /* Gemini Blue for headers */
+        margin-top: 0;
+        font-weight: 500;
+    }
+    .gemini-card strong {
+        color: #ffffff;
+    }
+    .gemini-card ul {
+        margin-left: 20px;
+    }
     </style>
     """, unsafe_allow_html=True)
 
-def generate_medical_pdf(report_text, genomic_risks, qr_data):
-    """Generates a clinical PDF with QR code and electronic signature."""
+# --- PDF GENERATION ENGINE ---
+def generate_signed_pdf(patient_text, genomics_text, qr_data):
     buffer = io.BytesIO()
-    p = canvas.Canvas(buffer, pagesize=A4)
+    c = canvas.Canvas(buffer, pagesize=A4)
     width, height = A4
     
-    # Header
-    p.setFont("Helvetica-Bold", 16)
-    p.drawString(1*inch, height - 1*inch, "ABYSSINIA INTELLIGENCE | PRECISION REPORT")
-    p.setFont("Helvetica", 10)
-    p.drawString(1*inch, height - 1.2*inch, "Date: 2026-02-12 | System Version: 8.0.2")
-    p.line(1*inch, height - 1.3*inch, 7.2*inch, height - 1.3*inch)
-
-    # Content
-    text_object = p.beginText(1*inch, height - 1.7*inch)
-    text_object.setFont("Helvetica", 11)
-    text_object.setLeading(14)
+    # 1. Header & Logo Placeholder
+    c.setFillColorRGB(0.1, 0.1, 0.1) # Dark Text
+    c.setFont("Helvetica-Bold", 16)
+    c.drawString(1*inch, height - 1*inch, "ABYSSINIA INTELLIGENCE | PRECISION REPORT")
     
-    # Split text to fit page
-    lines = report_text.split('\n')
-    for line in lines[:30]: # Limit lines for one page demo
-        text_object.textLine(line)
-    p.drawText(text_object)
+    c.setFillColorRGB(0.4, 0.4, 0.4)
+    c.setFont("Helvetica", 10)
+    c.drawString(1*inch, height - 1.25*inch, f"Date: 2026-02-12 | Authorized By: OMNI V8 Orchestrator")
+    c.line(1*inch, height - 1.4*inch, 7.27*inch, height - 1.4*inch)
 
-    # QR Code Generation
+    # 2. Body Text (Simple wrap for demo)
+    text_obj = c.beginText(1*inch, height - 1.8*inch)
+    text_obj.setFont("Helvetica", 11)
+    text_obj.setFillColorRGB(0, 0, 0)
+    text_obj.setLeading(14)
+    
+    # Merging content for the PDF
+    full_content = f"CLINICAL SUMMARY:\n{patient_text}\n\nGENOMIC RISK FACTORS:\n{genomics_text}"
+    
+    # Basic text wrapping (ReportLab usually requires Platypus for advanced wrapping, 
+    # but this keeps it single-file simple)
+    lines = full_content.split('\n')
+    line_limit = 45
+    for line in lines[:line_limit]: 
+        text_obj.textLine(line[:90]) # Truncate long lines to fit width
+    if len(lines) > line_limit:
+        text_obj.textLine("... [Content Truncated for One-Page Summary] ...")
+        
+    c.drawText(text_obj)
+
+    # 3. QR Code & Electronic Signature
+    # Generate QR Image
     qr = qrcode.make(qr_data)
-    qr_path = "temp_qr.png"
-    qr.save(qr_path)
+    qr_buffer = io.BytesIO()
+    qr.save(qr_buffer, format="PNG")
+    qr_buffer.seek(0)
+    qr_img = ImageReader(qr_buffer)
+
+    # Draw Footer Line
+    c.setStrokeColorRGB(0.8, 0.8, 0.8)
+    c.line(1*inch, 2*inch, 7.27*inch, 2*inch)
+
+    # Draw QR
+    c.drawImage(qr_img, 1*inch, 0.7*inch, width=1.2*inch, height=1.2*inch)
+
+    # Draw Signature Text
+    c.setFont("Helvetica-Bold", 12)
+    c.setFillColorRGB(0, 0, 0)
+    c.drawString(2.5*inch, 1.5*inch, "ELECTRONICALLY SIGNED")
     
-    # Footer: Signature and QR
-    p.line(1*inch, 2.5*inch, 7.2*inch, 2.5*inch)
-    p.drawImage(qr_path, 1*inch, 1*inch, width=1*inch, height=1*inch)
+    c.setFont("Courier", 9)
+    c.setFillColorRGB(0.3, 0.3, 0.3)
+    c.drawString(2.5*inch, 1.3*inch, f"Hash: {hash(full_content)}")
+    c.drawString(2.5*inch, 1.15*inch, "Verification: https://abyssinia.ai/verify")
     
-    p.setFont("Helvetica-Bold", 12)
-    p.drawString(4*inch, 1.8*inch, "ELECTRONIC SIGNATURE")
-    p.setFont("Helvetica-Oblique", 10)
-    p.drawString(4*inch, 1.6*inch, "Digitally Verified by OMNI Orchestrator V8")
-    p.drawString(4*inch, 1.4*inch, "ID: AB-2026-X9912")
-
-    p.showPage()
-    p.save()
+    c.save()
     buffer.seek(0)
     return buffer
 
-# Initialize Agents
+# --- INITIALIZATION ---
 orchestrator = OMNIOrchestratorV8(st.secrets["GEMINI_API_KEY"])
 genomics_engine = OMNIGenomics()
 
-# Sidebar
-with st.sidebar:
-    st.title("Clinical Workspace")
-    audio_file = st.audio_input("Record Encounter")
-    vcf_upload = st.file_uploader("Genomic Data", type=['vcf', 'txt'])
+st.title("Abyssinia Intelligence V8")
+st.caption("Ambient Intelligence & Pharmacogenomics | System Status: ONLINE")
 
-# Main UI
-st.title("Digital Twin Orchestrator")
-col_left, col_right = st.columns([1.2, 0.8])
+col_ambient, col_genomics = st.columns([1, 1])
 
-if st.button("Generate Digital Twin & Sign"):
-    if audio_file and vcf_upload:
-        with st.spinner("Analyzing and Signing Report..."):
-            # 1. AI Logic
+with col_ambient:
+    st.subheader("Ambient Scribe")
+    audio_file = st.audio_input("Record Patient Encounter")
+    
+    if audio_file:
+        with st.spinner("Scribing..."):
             soap_note = orchestrator.scribe_audio(audio_file.getvalue())
+            st.session_state.soap_note = soap_note
+            # UPDATED: Cleaner display than code block
+            st.info(f"Transcript Processed: {len(soap_note)} chars")
+
+with col_genomics:
+    st.subheader("Genomic Mapping")
+    vcf_upload = st.file_uploader("Upload Genomic (VCF) Data", type=['vcf', 'txt'])
+    
+    if vcf_upload:
+        # Simulated VCF analysis
+        vcf_data = pd.DataFrame({
+            "Chromosome": np.random.randint(1, 23, 50),
+            "Position": np.random.randint(1000, 1000000, 50),
+            "Risk_Score": np.random.rand(50),
+            "Clinical_Significance": np.random.randint(1, 10, 50)
+        })
+        st.plotly_chart(genomics_engine.plot_variant_density(vcf_data))
+
+if st.button("Generate Precision Synthesis"):
+    if 'soap_note' in st.session_state and vcf_upload:
+        with st.spinner("Orchestrating Multi-Agent Swarm..."):
             genomic_risks = orchestrator.map_genomics(vcf_upload.getvalue().decode())
-            final_report = orchestrator.synthesize_digital_twin(soap_note, genomic_risks)
             
-            # 2. Store State
-            st.session_state.final_report = final_report
-            st.session_state.genomic_risks = genomic_risks
+            # Using the Corrected V8 Method Name
+            final_report = orchestrator.synthesize_digital_twin(st.session_state.soap_note, genomic_risks)
             
-            # 3. Create JSON-LD
-            digital_twin = {"@type": "MedicalReport", "content": final_report}
-            st.session_state.json_ld = json.dumps(digital_twin)
-            
-            # 4. Generate PDF
-            qr_link = "https://abyssinia.ai/verify/AB-2026-X9912"
-            st.session_state.pdf_buffer = generate_medical_pdf(final_report, genomic_risks, qr_link)
-
-if 'final_report' in st.session_state:
-    with col_left:
-        st.subheader("Clinical Synthesis")
-        st.markdown(f'<div class="gemini-response">{st.session_state.final_report}</div>', unsafe_allow_html=True)
-        
-        st.divider()
-        
-        # New Feature: Signed PDF Export
-        st.download_button(
-            label="Download Signed PDF Report",
-            data=st.session_state.pdf_buffer,
-            file_name="Signed_Medical_Report_V8.pdf",
-            mime="application/pdf"
-        )
-        
-        st.download_button(
-            label="Export JSON-LD",
-            data=st.session_state.json_ld,
-            file_name="digital_twin.json",
-            mime="application/ld+json"
-        )
-
-    with col_right:
-        st.subheader("Genomic Risk Landscape")
-        mock_df = pd.DataFrame({"Chromosome": np.random.randint(1, 23, 15), "Risk": np.random.rand(15)})
-        st.plotly_chart(genomics_engine.plot_variant_density(mock_df), use_container_width=True)
-        st.info(st.session_state.genomic_risks)
+            # 1. DISPLAY: Gemini-Style Card (Markdown with CSS class)
+            st.markdown(f"""
+            <div class="gemini-card">
+            {final_report}
+            </div>
+            """, unsafe_allow_html=True)
+
+            # 2. GENERATE PDF: With Signature & QR
+            pdf_data = generate_signed_pdf(
+                st.session_state.soap_note, 
+                genomic_risks, 
+                "Abyssinia-V8-Patient-ID-5501"
+            )
+
+            # 3. EXPORT BUTTON
+            st.download_button(
+                label="Download Signed Clinical Report (PDF)",
+                data=pdf_data,
+                file_name="Abyssinia_Signed_Report.pdf",
+                mime="application/pdf"
+            )
+
+    else:
+        st.error("Incomplete Data: V8 requires both Ambient Scribe and Genomic Data.")