Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import speech_recognition as sr
 import time
 import difflib
-import random
+import os
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 from happytransformer import HappyTextToText, TTSettings
 
@@ -17,7 +17,6 @@ CSS = """
 .diff-del { color: #ef4444; background: #fee2e2; padding: 2px 4px; border-radius: 4px; }
 """
 
-# Three.js Template
 THREEJS_TEMPLATE = """
 <div id="container"></div>
 <script async src="https://unpkg.com/[email protected]/dist/es-module-shims.js"></script>
@@ -100,7 +99,6 @@ window.updateGrammarVisuals = (score) => {
 };
 </script>
 """
-
 # Load models
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
@@ -136,25 +134,40 @@ def analyze_grammar(text):
     }
 
 def process_input(audio_path, text):
-    if audio_path:
-        recognizer = sr.Recognizer()
-        with sr.AudioFile(audio_path) as source:
-            audio = recognizer.record(source)
+    # Handle audio input
+    if audio_path and os.path.exists(audio_path):
         try:
+            recognizer = sr.Recognizer()
+            with sr.AudioFile(audio_path) as source:
+                audio = recognizer.record(source)
             text = recognizer.recognize_google(audio)
-        except sr.UnknownValueError:
-            return "Could not understand audio", 0, "", ""
+        except Exception as e:
+            return [
+                "Audio processing error", 
+                0, 
+                f"<span style='color:red'>Error: {str(e)}</span>",
+                "<script>window.updateGrammarVisuals(0)</script>"
+            ]
     
+    # Handle text input
     if not text.strip():
-        return "No input", 0, "", ""
+        return ["No input provided", 0, "", "<script>window.updateGrammarVisuals(0)</script>"]
     
-    results = analyze_grammar(text)
-    return [
-        results['original'],
-        results['score'],
-        results['diff_html'],
-        f"<script>window.updateGrammarVisuals({results['score']})</script>"
-    ]
+    try:
+        results = analyze_grammar(text)
+        return [
+            results['original'],
+            results['score'],
+            results['diff_html'],
+            f"<script>window.updateGrammarVisuals({results['score']})</script>"
+        ]
+    except Exception as e:
+        return [
+            "Analysis error", 
+            0, 
+            f"<span style='color:red'>Error: {str(e)}</span>",
+            "<script>window.updateGrammarVisuals(0)</script>"
+        ]
 
 with gr.Blocks(css=CSS) as app:
     gr.Markdown("""
@@ -182,17 +195,18 @@ with gr.Blocks(css=CSS) as app:
             diff_output = gr.HTML(label="📝 Text Corrections")
             hidden_trigger = gr.HTML(visible=False)
 
+    # Fixed examples configuration
     gr.Markdown("### Example Sentences")
     gr.Examples(
         examples=[
-            ["I is going to the park yesterday."],
+            ["I is going to the park yesterday."],  # Text-only examples
             ["Their happy about there test results."],
             ["She dont like apples, but she like bananas."]
         ],
-        inputs=[text_input],
+        inputs=[text_input],  # Only text input
         outputs=[text_input, grammar_score, diff_output, hidden_trigger],
-        fn=process_input,
-        cache_examples=True
+        fn=lambda text: process_input(None, text),  # Explicitly handle text-only examples
+        cache_examples=False  # Disable caching to prevent startup issues
     )
     
     submit_btn.click(
@@ -211,6 +225,5 @@ if __name__ == "__main__":
     app.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=True,
-        favicon_path="https://raw.githubusercontent.com/gradio-app/gradio/main/guides/assets/logo.png"
+        share=False  # Disable sharing until basic functionality works
     )