deploy login

.gitignore

@@ -1,2 +1,3 @@
 .env
-__pycache__
+__pycache__
+flagged

app.py

@@ -1,4 +1,6 @@
 import os
+import json
+import re
 import gradio as gr
 from google import genai
 from google.genai import types
@@ -20,7 +22,7 @@ def audio_from_bytes(audio_file_path: str):
         return f"An error occurred: {e}"
 
 
-def transcribe_and_summarize(audio_file):
+def transcribe_and_summarize(audio_file, session):
     """Processes audio with Gemini API and returns a SOAP summary."""
     if audio_file is None:
         return "No audio file uploaded."
@@ -71,23 +73,143 @@ Identifikasi dan berikan saran dalam bahasa Indonesia tindakan logis selanjutnya
 
     # Process the audio
     response_text = ""
-    for chunk in client.models.generate_content_stream(
-        model=model,
-        contents=contents,
-        config=generate_content_config,
-    ):
-        response_text += chunk.text
-
-    return response_text
-
-
-# Gradio UI
-demo = gr.Interface(
-    fn=transcribe_and_summarize,
-    inputs=gr.Audio(type="filepath"),
-    outputs="text",
-    title="eH Medical Transcription",
-    description="Upload an Indonesian audio file of a doctor-patient conversation and get a SOAP summary.",
-)
+    # for chunk in client.models.generate_content_stream(
+    #     model=model,
+    #     contents=contents,
+    #     config=generate_content_config,
+    # ):
+    #     response_text += chunk.text
+    counter_display, session = click_button(session)
+    return response_text, counter_display, session
+
+
+DATA_FILE = "user_data.json"
+
+
+def load_user_data():
+    """Load user data from JSON file."""
+    if os.path.exists(DATA_FILE):
+        with open(DATA_FILE, "r") as file:
+            return json.load(file)
+    return {}  # Return empty dictionary if no data
+
+
+def save_user_data(username, email):
+    """Save user data to JSON file with default counter = 10 if new."""
+    data = load_user_data()
+
+    if username not in data:  # New user
+        data[username] = {"email": email, "counter": 10}  # Set default counter to 10
+
+    with open(DATA_FILE, "w") as file:
+        json.dump(data, file, indent=4)
+
+    return data
+
+
+def is_valid_wa(username):
+    """Check if username is all numbers, at least 11 characters, and starts with 08 or 62."""
+    return re.fullmatch(r"^(08|62)\d{9,}$", username) is not None
+
+
+def is_valid_email(email):
+    """Check if email format is valid."""
+    return re.fullmatch(r"^[\w\.-]+@[\w\.-]+\.\w+$", email) is not None
+
+
+def login(username, email, session):
+    """Handles user login or registration with validation."""
+
+    if not is_valid_wa(username):
+        return (
+            "❌ Invalid WA! Nomor WA minimal 12 digits dimulai '08' atau '62'.",
+            session,
+            gr.update(visible=True),
+            gr.update(visible=False),
+        )
+
+    if not is_valid_email(email):
+        return (
+            "❌ Invalid email format! Please enter a valid email.",
+            session,
+            gr.update(visible=True),
+            gr.update(visible=False),
+        )
+
+    data = save_user_data(username, email)  # Save or retrieve user data
+
+    session["username"] = username
+    session["counter"] = data[username]["counter"]
+
+    return "", session, gr.update(visible=False), gr.update(visible=True)
+
+
+def click_button(session):
+    """Decrease counter on button click."""
+    if session["counter"] > 0:
+        session["counter"] -= 1
+
+        # Update the user data in JSON file
+        data = load_user_data()
+        data[session["username"]]["counter"] = session["counter"]
+
+        with open(DATA_FILE, "w") as file:
+            json.dump(data, file, indent=4)
+
+        if session["counter"] == 0:
+            return (
+                f"⚠️ Quota habis. Silahkan mengunjungi https://ehealth.co.id atau WA 6285777779926",
+                session,
+            )
+        return f"Counter: {session['counter']}", session
+    else:
+        return (
+            f"⚠️ Quota habis. Silahkan mengunjungi https://ehealth.co.id atau WA 6285777779926",
+            session,
+        )
+
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    session = gr.State({"username": None, "counter": 0})  # Manage session state
+
+    gr.Markdown("# 🛡️ eHealth.co.id Scribe")
+
+    # Login Section
+    login_block = gr.Column(visible=True)
+    with login_block:
+        email_input = gr.Textbox(label="Email")
+        username_input = gr.Textbox(label="WA", type="password")  # Hide input
+        login_button = gr.Button("🔑 Login / Register")
+        gr.Markdown(
+            """### dengan login saya menyetujui ketentuan penggunaan data eHealth.co.id dan tidak akan menuntut perusahaan eHealth.co.id untuk uji coba gratis AI Transcribe ini
+            
+            saya memahami bahwa tidak ada data suara maupun tulisan medis yang akan disimpan oleh perusahaan 
+            
+            seluruh data yang saya sediakan adalah data yang benar dan tidak bermaksud melanggar hukum
+            
+            ## Setelah quota habis, Silahkan mengunjungi https://ehealth.co.id atau WA 6285777779926"""
+        )
+        output_text = gr.Textbox(label="Status", interactive=False)
+
+    # Main User Interface (After Login)
+    user_block = gr.Column(visible=False)
+    with user_block:
+        counter_display = gr.Textbox(label="Status Message", interactive=False)
+        gr.Interface(
+            fn=transcribe_and_summarize,
+            inputs=[gr.Audio(type="filepath", sources="microphone"), session],
+            outputs=["text", counter_display, session],
+            title="eH Transcription",
+            description="pastikan HP/Laptop memiliki microphone untuk merekam percakapan dokter-pasien menjadi rekam medis SOAP",
+            allow_flagging="never",
+        )
+
+        # Login button action
+        login_button.click(
+            login,
+            [username_input, email_input, session],
+            [output_text, session, login_block, user_block],
+        )
 
 demo.launch()

user_data.json

@@ -0,0 +1,6 @@
+{
+    "6285777779926": {
+        "email": "[email protected]",
+        "counter": 10
+    }
+}