Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 
-# Preload models (optimized and corrected)
+# Preload models
 summarizer = pipeline("summarization")
 sentiment_analyzer = pipeline("sentiment-analysis", model="cardiffnlp/twitter-roberta-base-sentiment")
 translator_hi = pipeline("translation", model="Helsinki-NLP/opus-mt-en-hi")
@@ -10,10 +10,9 @@ translator_de = pipeline("translation", model="Helsinki-NLP/opus-mt-en-de")
 translator_es = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
 translator_ta = pipeline("translation", model="facebook/nllb-200-distilled-600M", src_lang="eng_Latn", tgt_lang="tam_Taml")
 speech_to_text = pipeline("automatic-speech-recognition", model="openai/whisper-small")
-question_generator = pipeline("text2text-generation", model="valhalla/t5-base-e2e-qg")
-
-# --- Functional Modules ---
+question_generator = pipeline("e2e-qg")
 
+# Functional logic
 def summarize(text):
     return summarizer(text, max_length=60, min_length=20, do_sample=False)[0]['summary_text']
 
@@ -45,87 +44,70 @@ def transcribe(audio):
     return speech_to_text(audio)["text"]
 
 def generate_questions(text):
-    output = question_generator("generate questions: " + text)
-    return output[0]['generated_text']
-
-# --- UI Sections for each task ---
-
-def summarization_ui():
-    with gr.Column():
-        input_text = gr.Textbox(label="Enter a long paragraph", lines=12, placeholder="Paste your paragraph here...")
-        output_text = gr.Textbox(label="Summarized text", lines=4)
-        summarize_btn = gr.Button("Summarize")
-        summarize_btn.click(summarize, inputs=input_text, outputs=output_text)
-
-def sentiment_ui():
-    with gr.Column():
-        input_text = gr.Textbox(label="Enter a sentence", lines=3)
-        output_text = gr.Textbox(label="Sentiment")
-        analyze_btn = gr.Button("Analyze Sentiment")
-        analyze_btn.click(analyze_sentiment, inputs=input_text, outputs=output_text)
-
-def translation_ui():
-    with gr.Column():
-        input_text = gr.Textbox(label="Enter English text", lines=4)
-        lang_dropdown = gr.Dropdown(["Tamil", "Hindi", "French", "German", "Spanish"], value="Tamil", label="Target Language")
-        output_text = gr.Textbox(label="Translated text", lines=4)
-        translate_btn = gr.Button("Translate")
-        translate_btn.click(translate, inputs=[input_text, lang_dropdown], outputs=output_text)
-
-def speech_ui():
-    with gr.Column():
-        audio = gr.Audio(source="microphone", type="filepath", label="Record or Upload")
-        output_text = gr.Textbox(label="Recognized Text")
-        speech_btn = gr.Button("Convert Speech to Text")
-        speech_btn.click(transcribe, inputs=audio, outputs=output_text)
-
-def question_ui():
-    with gr.Column():
-        input_text = gr.Textbox(label="Enter a paragraph", lines=10)
-        output_text = gr.Textbox(label="Generated Questions", lines=10)
-        question_btn = gr.Button("Generate Questions")
-        question_btn.click(generate_questions, inputs=input_text, outputs=output_text)
-
-# --- App Layout ---
-
-def build_interface():
-    with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown("""
-            <h1 style='text-align:center;'>🌐 Multi-Task AI App (Hugging Face Space)</h1>
-            <h4 style='text-align:center;'>By Joel — Powered by Transformers</h4>
-        """)
-
-        with gr.Row():
-            btn1 = gr.Button("Text Summarization")
-            btn2 = gr.Button("Sentiment Analysis")
-            btn3 = gr.Button("Translation")
-            btn4 = gr.Button("Speech-to-Text")
-            btn5 = gr.Button("Question Generation")
-
-        container = gr.Column()
-
-        def update_view(tab_name):
-            container.children.clear()
-            with container:
-                if tab_name == "summarization":
-                    summarization_ui()
-                elif tab_name == "sentiment":
-                    sentiment_ui()
-                elif tab_name == "translation":
-                    translation_ui()
-                elif tab_name == "speech":
-                    speech_ui()
-                elif tab_name == "question":
-                    question_ui()
-
-        btn1.click(lambda: update_view("summarization"), outputs=[])
-        btn2.click(lambda: update_view("sentiment"), outputs=[])
-        btn3.click(lambda: update_view("translation"), outputs=[])
-        btn4.click(lambda: update_view("speech"), outputs=[])
-        btn5.click(lambda: update_view("question"), outputs=[])
-
-    return demo
-
-demo = build_interface()
-demo.launch()
+    output = question_generator(text)
+    return "\n".join(f"- {item['question']}" for item in output[:10])
+
+# UI App
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 🌐 AI Buddy — Multi-task App by Joel")
+    gr.Markdown("Choose a task below:")
+
+    # Navigation buttons
+    with gr.Row():
+        task = gr.State(value="")  # to keep track of current screen
+        btn_summ = gr.Button("Text Summarization")
+        btn_sent = gr.Button("Sentiment Analysis")
+        btn_trans = gr.Button("Translation")
+        btn_speech = gr.Button("Speech-to-Text")
+        btn_qgen = gr.Button("Question Generation")
+
+    output_container = gr.Column(visible=False)
+
+    # Summarization
+    with gr.Column(visible=False) as summarize_tab:
+        text = gr.Textbox(label="Enter paragraph", lines=10)
+        out = gr.Textbox(label="Summary", lines=4)
+        gr.Button("Summarize").click(summarize, inputs=text, outputs=out)
+
+    # Sentiment
+    with gr.Column(visible=False) as sentiment_tab:
+        sent_in = gr.Textbox(label="Enter sentence", lines=3)
+        sent_out = gr.Textbox(label="Sentiment")
+        gr.Button("Analyze Sentiment").click(analyze_sentiment, inputs=sent_in, outputs=sent_out)
+
+    # Translation
+    with gr.Column(visible=False) as translate_tab:
+        tran_in = gr.Textbox(label="Enter English text", lines=3)
+        lang = gr.Dropdown(["Tamil", "Hindi", "French", "German", "Spanish"], value="Tamil", label="Language")
+        tran_out = gr.Textbox(label="Translated text", lines=3)
+        gr.Button("Translate").click(translate, inputs=[tran_in, lang], outputs=tran_out)
+
+    # Speech-to-Text
+    with gr.Column(visible=False) as speech_tab:
+        audio = gr.Audio(source="microphone", type="filepath", label="Record or upload")
+        speech_out = gr.Textbox(label="Recognized Text")
+        gr.Button("Convert Speech to Text").click(transcribe, inputs=audio, outputs=speech_out)
+
+    # Question Generation
+    with gr.Column(visible=False) as question_tab:
+        ques_in = gr.Textbox(label="Enter a paragraph", lines=10)
+        ques_out = gr.Textbox(label="Generated Questions", lines=10)
+        gr.Button("Generate Questions").click(generate_questions, inputs=ques_in, outputs=ques_out)
+
+    # Logic to show/hide tabs
+    def show_tab(tab_name):
+        return [
+            gr.update(visible=(tab_name == "summarize")),
+            gr.update(visible=(tab_name == "sentiment")),
+            gr.update(visible=(tab_name == "translate")),
+            gr.update(visible=(tab_name == "speech")),
+            gr.update(visible=(tab_name == "question")),
+        ]
+
+    btn_summ.click(show_tab, outputs=[summarize_tab, sentiment_tab, translate_tab, speech_tab, question_tab], inputs=[], _js="() => 'summarize'")
+    btn_sent.click(show_tab, outputs=[summarize_tab, sentiment_tab, translate_tab, speech_tab, question_tab], inputs=[], _js="() => 'sentiment'")
+    btn_trans.click(show_tab, outputs=[summarize_tab, sentiment_tab, translate_tab, speech_tab, question_tab], inputs=[], _js="() => 'translate'")
+    btn_speech.click(show_tab, outputs=[summarize_tab, sentiment_tab, translate_tab, speech_tab, question_tab], inputs=[], _js="() => 'speech'")
+    btn_qgen.click(show_tab, outputs=[summarize_tab, sentiment_tab, translate_tab, speech_tab, question_tab], inputs=[], _js="() => 'question'")
 
+demo.launch()