Update app.py

app.py

@@ -4,108 +4,108 @@ from datetime import datetime
 from google import genai
 from google.genai import types
 
-# --- 1. PAGE CONFIG & STYLING ---
-st.set_page_config(page_title="Abyssinia Scribe Pro", page_icon="📝", layout="wide")
+# --- 1. PAGE CONFIG & MODERN CSS ---
+st.set_page_config(page_title="Abyssinia Scribe", page_icon="🏥", layout="wide")
 
-# ChatGPT-Style Dark Mode Theme for the Output Note
+# Custom CSS for "ChatGPT Style" Output - Clean Sans-Serif & Modern Spacing
 st.markdown("""
     <style>
-    .chatgpt-note {
-        background-color: #0d1117;
-        color: #c9d1d9;
-        padding: 30px;
-        border-radius: 10px;
-        font-family: 'Fira Code', 'Courier New', monospace;
-        line-height: 1.7;
-        border: 1px solid #30363d;
-        white-space: pre-wrap;
-        box-shadow: 0 10px 30px rgba(0,0,0,0.5);
+    /* ChatGPT-style Container */
+    .chat-gpt-style {
+        background-color: #ffffff;
+        color: #343541;
+        padding: 40px;
+        border-radius: 12px;
+        font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
+        line-height: 1.6;
+        border: 1px solid #e5e5e5;
+        box-shadow: 0 4px 24px rgba(0,0,0,0.08);
+        max-width: 800px;
+        margin: auto;
     }
-    .scribe-sig {
-        margin-top: 25px;
-        padding-top: 15px;
-        border-top: 1px solid #30363d;
-        color: #8b949e;
+    /* Clinical Headers */
+    .note-section-header {
+        color: #10a37f; /* ChatGPT Green accent */
+        font-weight: 600;
+        font-size: 1.1em;
+        margin-top: 20px;
+        margin-bottom: 8px;
+        text-transform: uppercase;
+        letter-spacing: 0.5px;
+    }
+    /* Accurate Signature */
+    .clinical-sig {
+        margin-top: 40px;
+        padding-top: 20px;
+        border-top: 1px solid #f0f0f0;
+        color: #6e6e80;
         font-size: 0.9em;
-        font-style: italic;
     }
     </style>
     """, unsafe_allow_html=True)
 
 # API Initialization
-api_key = os.environ.get("GOOGLE_API_KEY")
-client = genai.Client(api_key=api_key)
+client = genai.Client(api_key=os.environ.get("GOOGLE_API_KEY"))
 
-# --- 2. APP INTERFACE ---
+# --- 2. THE UI ---
 st.title("🏥 Abyssinia Intelligence Scribe")
-st.markdown("Professional Clinical Documentation & SOAP Note Engine")
 
 col1, col2 = st.columns([1, 1], gap="large")
 
 with col1:
-    st.subheader("🎙️ Capture Consultation")
-    
-    # Microphone Recording Component
-    audio_file = st.audio_input("Record Live Encounter")
+    st.subheader("🎙️ Input Consultation")
+    audio_file = st.audio_input("Record Conversation")
+    text_backup = st.text_area("Or type/paste notes here:", height=250)
     
-    # Manual Text Backup
-    text_input = st.text_area("Or paste consultation notes here:", height=150)
-    
-    generate_btn = st.button("Generate Final SOAP Note", type="primary")
+    generate_btn = st.button("Finalize Scribe Report", type="primary")
 
 with col2:
-    st.subheader("📄 Clinical Record")
+    st.subheader("📝 Final Medical Note")
     
     if generate_btn:
-        input_parts = []
-        
-        # FIX: Correctly formatting the input for the 2026 SDK
+        # Prepare content parts
+        parts = []
         if audio_file:
-            # We read the raw bytes from the Streamlit UploadedFile object
-            audio_bytes = audio_file.read()
-            input_parts.append(
-                types.Part.from_bytes(data=audio_bytes, mime_type="audio/wav")
-            )
-        elif text_input:
-            input_parts.append(types.Part.from_text(text=text_input))
-        
-        if not input_parts:
-            st.warning("Please provide audio or text input.")
+            parts.append(types.Part.from_bytes(data=audio_file.read(), mime_type="audio/wav"))
+        elif text_backup:
+            parts.append(types.Part.from_text(text=text_backup))
+            
+        if not parts:
+            st.error("Error: No consultation data detected.")
         else:
-            with st.spinner("Abyssinia AI is generating the clinical record..."):
+            with st.spinner("Generating document..."):
                 try:
-                    # Gemini 3.0 Flash Processing
+                    # Request structured data from Gemini 3.0 Flash
                     response = client.models.generate_content(
                         model="gemini-3-flash-preview",
                         config=types.GenerateContentConfig(
-                            system_instruction="Create a formal medical SOAP note. Use professional formatting.",
-                            temperature=0.1 # Very low for high accuracy
+                            system_instruction="Create a detailed, structured medical SOAP note. Format sections clearly.",
+                            temperature=0.2
                         ),
-                        contents=input_parts
+                        contents=parts
                     )
                     
-                    # 3. Accurate Signature Date
+                    # Formatting text to look "ChatGPT attractive" rather than code-like
+                    formatted_output = response.text.replace("### ", "<div class='note-section-header'>")
+                    formatted_output = formatted_output.replace("\n", "<br>")
+                    
+                    # Get Local Ethiopia/Server Time
                     now = datetime.now()
-                    formatted_date = now.strftime("%A, %B %d, %Y")
-                    formatted_time = now.strftime("%I:%M %p")
+                    timestamp = now.strftime("%d %B %Y at %H:%M")
                     
-                    # Constructing the "Attractive" HTML output
+                    # Rendering the HTML
                     st.markdown(f"""
-                        <div class="chatgpt-note">
-{response.text}
-<div class="scribe-sig">
-— Electronically generated by Abyssinia Intelligence Scribe<br>
-— Timestamp: {formatted_date} at {formatted_time}<br>
-— Reference: ABY-{now.strftime('%H%M%S')}
-</div>
+                        <div class="chat-gpt-style">
+                            {formatted_output}
+                            <div class="clinical-sig">
+                                <b>Digitally Authenticated:</b> Abyssinia AI Scribe Engine<br>
+                                <b>Finalized on:</b> {timestamp}<br>
+                                <b>Reference ID:</b> ABY-{now.strftime('%m%d-%H%M')}
+                            </div>
                         </div>
                     """, unsafe_allow_html=True)
                     
-                    st.download_button("Download Document", response.text, file_name=f"Clinical_Note_{now.strftime('%Y%m%d')}.txt")
-
+                    st.download_button("Export Clinical Note", response.text)
+                    
                 except Exception as e:
-                    st.error(f"Scribe Error: {e}")
-
-# --- 3. FOOTER ---
-st.markdown("---")
-st.caption("Abyssinia Intelligence | 2026 Healthcare Scribe Edition")
+                    st.error(f"Scribe Error: {e}")