Update app.py

app.py

@@ -119,7 +119,7 @@ def main():
         "Escribe tus términos de búsqueda o haz una pregunta terminando con el caracter ?:"
     )
 
-    all_results = pd.DataFrame(columns=["answer","score","start","end"])
+    all_results = pd.DataFrame(columns=["id","answer","score","start","end"])
     
     if query:
 
@@ -152,7 +152,7 @@ def main():
                 
                answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))
 
-               all_results.loc[i] = answer, max(outputs.start_logits.numpy()[0]), 0, 0
+               all_results.loc[i] = i,answer, max(outputs.start_logits.numpy()[0]), 0, 0
                            
                # Barra de progreso
                if (usar_barra_progreso==1):
@@ -167,6 +167,7 @@ def main():
            for index, row in all_results.iterrows():
                if (len(row['answer'])>0):    
                    cantidad_respuestas = cantidad_respuestas + 1
+                   i=row['id']
                    df_answer.loc[i, "answer"] = row.loc['answer']
                    df_answer.loc[i, "score"]= row.loc['score']
                    lista_noticias_respuestas.append(df_answer.loc[i].to_frame().T)