Update app.py

app.py

@@ -2,15 +2,21 @@ import torch
 import gradio as gr
 from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
 
-model_name = "deepset/roberta-base-squad2"
-
-pipe = pipeline('question-answering', model=model_name, tokenizer=model_name)
-
-model = AutoModelForQuestionAnswering.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-
-# context = "I am gabriel"
-# question = "what is my name?"
+model_eng   = "deepset/roberta-base-squad2"
+model_br    = "mrm8488/bert-base-portuguese-cased-finetuned-squad-v1-pt"
+
+PIPELINES = {
+    "English": pipeline(
+        "question-answering",
+        model="deepset/roberta-base-squad2",
+        tokenizer="deepset/roberta-base-squad2"
+    ),
+    "Português": pipeline(
+        "question-answering",
+        model="mrm8488/bert-base-portuguese-cased-finetuned-squad-v1-pt",
+        tokenizer="mrm8488/bert-base-portuguese-cased-finetuned-squad-v1-pt"
+    )
+}
 
 def read_file(file_content):
     try:
@@ -19,7 +25,7 @@ def read_file(file_content):
     except Exception as e:
         return f"Ocorreu um erro: {e}"
     
-def get_answer(text_content, question, file_content):
+def get_answer(lang , text_content, question, file_content):
     if text_content and file_content:
         return "Selecione apenas uma opção!"
     
@@ -31,6 +37,10 @@ def get_answer(text_content, question, file_content):
     else:
         content = text_content
 
+    pipe = PIPELINES.get(lang)
+    if pipe is None:
+        return "Idioma não suportado."
+    
     res = pipe(
         context=content,
         question=question,
@@ -38,10 +48,16 @@ def get_answer(text_content, question, file_content):
     return res
 
 demo = gr.Interface(fn=get_answer,
-                    inputs=[gr.Textbox(label="Input your context", lines=4), gr.Textbox(label="Input your question", lines=4), gr.File(label="Uploadl your file")],
-                    outputs=[gr.Textbox(label="Answer", lines=3)],
-                    title="QnA for files, txt or essays",
-                    description="Esse programa em python foi feito para a disciplina de sistemas multimidia em 2025/1",)
+                    inputs=[
+                    gr.Radio(["English", "Português"], label="Selecione o idioma"),
+                    gr.Textbox(label="Contexto (texto livre)", placeholder="Digite seu texto aqui...", lines=6),
+                    gr.Textbox(label="Pergunta", placeholder="Digite sua pergunta aqui...", lines=2),
+                    gr.File(label="Ou faça upload de um arquivo .txt", file_types=['.txt'])
+                    ],
+                    outputs=gr.Textbox(label="Resposta"),
+                    title="QnA Multilíngue",
+                    description="Faça perguntas em inglês ou português. Carrega o modelo adequado automaticamente.",
+                    )
 
 if __name__ == "__main__":
     demo.launch()