Update app.py

app.py

@@ -1,33 +1,48 @@
 import subprocess
 import sys
 from transformers import T5ForConditionalGeneration, T5Tokenizer
-from translate import Translator as LibreTranslator
 import gradio as gr
 
-# Install required packages (without conflicts)
-required_packages = [
-    'transformers>=4.30.0',
-    'sentencepiece',
-    'translate>=3.6.1',  # Alternative to googletrans
-    'torch',
-    'gradio'
-]
+# Install required packages with error handling
+def install(package):
+    try:
+        subprocess.check_call([sys.executable, "-m", "pip", "install", package])
+    except subprocess.CalledProcessError:
+        print(f"Failed to install {package}, trying with --user flag")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "--user", package])
 
-def install_packages():
-    for package in required_packages:
-        try:
-            subprocess.check_call([sys.executable, "-m", "pip", "install", package])
-        except subprocess.CalledProcessError:
-            print(f"Failed to install {package}, trying with --user flag")
-            subprocess.check_call([sys.executable, "-m", "pip", "install", "--user", package])
+# Install all dependencies
+install('transformers')
+install('sentencepiece')
+install('torch')
+install('gradio')
 
-install_packages()
+# Try multiple translation backends
+translation_backend = None
+try:
+    install('translate')
+    from translate import Translator as LibreTranslator
+    translation_backend = "libretrans"
+except:
+    try:
+        install('googletrans==3.1.0a0')
+        from googletrans import Translator
+        translation_backend = "googletrans"
+    except:
+        print("Warning: No translation backend available")
 
 class GOTSummarizer:
     def __init__(self):
         self.tokenizer = T5Tokenizer.from_pretrained("t5-small")
         self.model = T5ForConditionalGeneration.from_pretrained("t5-small")
-        self.translator = LibreTranslator()
+        
+        # Initialize translator based on available backend
+        if translation_backend == "libretrans":
+            self.translator = LibreTranslator()
+        elif translation_backend == "googletrans":
+            self.translator = Translator()
+        else:
+            self.translator = None
     
     def summarize(self, text, max_length=150):
         inputs = self.tokenizer("summarize: " + text, 
@@ -43,8 +58,14 @@ class GOTSummarizer:
         return self.tokenizer.decode(outputs[0], skip_special_tokens=True)
     
     def translate(self, text, lang='hi'):
+        if not self.translator:
+            return "[Translation unavailable] " + text
+            
         try:
-            return self.translator.translate(text, to_lang=lang)
+            if translation_backend == "libretrans":
+                return self.translator.translate(text, to_lang=lang)
+            else:  # googletrans
+                return self.translator.translate(text, dest=lang).text
         except Exception as e:
             print(f"Translation error: {e}")
             return text
@@ -63,25 +84,18 @@ def process(text, lang='en'):
 interface = gr.Interface(
     fn=process,
     inputs=[
-        gr.Textbox(label="Game of Thrones Text", lines=10, 
-                 placeholder="Paste book chapter text here..."),
+        gr.Textbox(label="Game of Thrones Text", lines=10),
         gr.Dropdown(
-            label="Translation Language",
-            choices=[
-                ('English', 'en'),
-                ('Hindi', 'hi'),
-                ('Punjabi', 'pa'),
-                ('Tamil', 'ta'),
-                ('Bengali', 'bn')
-            ],
+            label="Language",
+            choices=['en', 'hi', 'pa', 'ta', 'bn'],
             value='en'
         )
     ],
     outputs=gr.Textbox(label="Summary"),
     title="ASOIAF Chapter Summarizer",
-    description="Summarizes Game of Thrones chapters with translation support"
+    description="Summarizes Game of Thrones chapters with optional translation"
 )
 
 # Launch the interface
 if __name__ == "__main__":
-    interface.launch(server_port=7860, share=True)
+    interface.launch()