Upload 2 files

app.py

@@ -16,7 +16,7 @@ def call_generate_news(subject, sites, min_words, max_words):
         return 'Erro: Máximo de palavras não informado', ''
     if sites == '':
         return 'Erro: Domínios para pesquisa não informados', ''
-    if max_words < min_words:
+    if int(max_words) < int(min_words):
         return 'Erro: Máximo de palavras menor que o mínimo de palavras'
 
     filtered_list_sites = []

main.py

@@ -40,7 +40,11 @@ def generate_news(subject, min_words, max_words, sites):
     obj_rag = rag.Rag(vectorstore, min_words, max_words)
     result_news = obj_rag.generate_text(subject)
     if type(result_news) == str:
-        return 'Erro: ' + result_news
+        if 'maximum context length' in result_news:
+            msg = 'Erro: Reduza o número máximo de palavras.'
+        else:
+            msg = 'Erro: ' + result_news
+        return msg
 
     print('\n\n' + '*' * 50 + '\n\n')
     print(result_news[0])