Update app.py

app.py

@@ -9,6 +9,7 @@ from huggingface_hub import HfApi
 from pypdf import PdfReader
 import docx
 import traceback
+from gtts import gTTS
 
 DATA_FILE = "legislation_rules.json"
 RULINGS_FILE = "rulings_log.json"
@@ -72,22 +73,31 @@ def process_rule_document(file_obj, act_name):
     if not text: return "Error: Could not extract text from rule document."
     return add_rule_manually(act_name, os.path.basename(file_obj.name), text, f"File: {os.path.basename(file_obj.name)}")
 
+def generate_tts(text):
+    """Generates an audio file from text using gTTS."""
+    if not text or len(text) < 5:
+        return None
+    try:
+        clean_text = text.replace("#", "").replace("*", "").replace("`", "")
+        tts = gTTS(text=clean_text[:1500], lang='en') # Limit to first 1500 chars for speed
+        filename = f"ruling_audio_{int(datetime.now().timestamp())}.mp3"
+        tts.save(filename)
+        return filename
+    except Exception as e:
+        print(f"TTS Error: {e}")
+        return None
+
 def rule_on_issue(issue_description, issue_file, llm_endpoint, llm_key, llm_model):
-    """
-    Framework 2.2: Regulatory Reasoning Engine with Document Intake.
-    """
-    # Combine text input and file input
     file_text = extract_text_from_any(issue_file)
     combined_issue = f"{issue_description}\n\n[EXTRACTED FROM DOCUMENT]:\n{file_text}".strip()
     
     if not combined_issue:
-        return "Error: No issue description or document provided.", ""
+        return "Error: No issue description or document provided.", "", None
 
     rules = load_data(DATA_FILE)
     if not rules:
-        return "Error: Dataset is empty. Add rules (CRA, FCA, etc.) first.", ""
+        return "Error: Dataset is empty. Add rules (CRA, FCA, etc.) first.", "", None
     
-    # Context window management
     context = "\n".join([f"[{r['act']} - {r['section_title']}]: {r['text'][:600]}..." for r in rules[:20]])
     
     prompt = f"""
@@ -115,6 +125,9 @@ def rule_on_issue(issue_description, issue_file, llm_endpoint, llm_key, llm_mode
         except Exception as e:
             ruling_text = f"Inference Error: {str(e)}"
     
+    # Generate Audio for the Ruling
+    audio_file = generate_tts(ruling_text)
+    
     # Log the ruling
     rulings_log = load_data(RULINGS_FILE)
     new_ruling = {
@@ -126,7 +139,7 @@ def rule_on_issue(issue_description, issue_file, llm_endpoint, llm_key, llm_mode
     rulings_log.append(new_ruling)
     save_data(rulings_log, RULINGS_FILE)
     
-    return ruling_text, f"Ruling Logged (ID: {new_ruling['ruling_hash'][:8]})"
+    return ruling_text, f"Ruling Logged (ID: {new_ruling['ruling_hash'][:8]})", audio_file
 
 def sync_all(token, dataset_id):
     if not token or not dataset_id: return "Error: Missing credentials."
@@ -161,6 +174,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 rule_btn = gr.Button("Generate Formal Ruling", variant="primary")
             with gr.Column():
                 ruling_out = gr.Markdown(label="Official Ruling")
+                audio_out = gr.Audio(label="Audible Ruling", interactive=False)
                 log_status = gr.Textbox(label="Audit Status")
 
     with gr.Tab("➕ Manage Rules"):
@@ -184,7 +198,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         s_btn = gr.Button("Sync Everything to Dataset")
         s_out = gr.Textbox(label="Sync Status")
 
-    rule_btn.click(fn=rule_on_issue, inputs=[issue_desc, issue_file, end, key, mod], outputs=[ruling_out, log_status])
+    rule_btn.click(fn=rule_on_issue, inputs=[issue_desc, issue_file, end, key, mod], outputs=[ruling_out, log_status, audio_out])
     doc_btn.click(fn=process_rule_document, inputs=[doc_in, doc_act], outputs=op_status)
     m_btn.click(fn=add_rule_manually, inputs=[m_act, m_tit, m_txt, m_src], outputs=op_status)
     refresh_btn.click(fn=view_stats, outputs=stats_out)