Update app.py

app.py

@@ -1,24 +1,16 @@
 import subprocess
 import sys
 from transformers import T5ForConditionalGeneration, T5Tokenizer
-from googletrans import Translator
 import gradio as gr
 
-# Install required packages
-def install(package):
-    subprocess.check_call([sys.executable, "-m", "pip", "install", package])
-
-install('transformers')
-install('sentencepiece')
-install('googletrans==4.0.0-rc1')
-install('torch')
+# Fix httpcore compatibility
+subprocess.check_call([sys.executable, "-m", "pip", "install", "httpcore==0.15.0", "googletrans==4.0.0-rc1"])
+from googletrans import Translator
 
 class GOTSummarizer:
     def __init__(self):
-        # Initialize model and tokenizer
-        model_name = "t5-small"
-        self.tokenizer = T5Tokenizer.from_pretrained(model_name)
-        self.model = T5ForConditionalGeneration.from_pretrained(model_name)
+        self.tokenizer = T5Tokenizer.from_pretrained("t5-small")
+        self.model = T5ForConditionalGeneration.from_pretrained("t5-small")
         self.translator = Translator()
     
     def summarize(self, text, max_length=150):
@@ -38,36 +30,20 @@ class GOTSummarizer:
             print(f"Translation error: {e}")
             return text
 
-# Initialize summarizer
 summarizer = GOTSummarizer()
 
 def process(text, lang='en'):
     if not text.strip():
-        return "Please enter some text to summarize"
-    
-    # Summarize the text
+        return "Please enter text to summarize"
     summary = summarizer.summarize(text)
-    
-    # Translate if needed
-    if lang != 'en':
-        return summarizer.translate(summary, lang)
-    return summary
+    return summary if lang == 'en' else summarizer.translate(summary, lang)
 
-# Create Gradio interface
-interface = gr.Interface(
+gr.Interface(
     fn=process,
     inputs=[
-        gr.Textbox(label="Input Text", lines=10, placeholder="Paste a Game of Thrones chapter text here..."),
-        gr.Dropdown(
-            label="Language",
-            choices=['en', 'hi', 'pa', 'ta', 'bn'],
-            value='en'
-        )
+        gr.Textbox(label="Input Text", lines=10),
+        gr.Dropdown(['en', 'hi', 'pa', 'ta', 'bn'], label="Language")
     ],
-    outputs=gr.Textbox(label="Summary"),
-    title="Game of Thrones Chapter Summarizer",
-    description="Summarizes chapters from ASOIAF books with translation support for Indian languages"
-)
-
-# Launch the interface
-interface.launch()
+    outputs="text",
+    title="Game of Thrones Summarizer"
+).launch()