bishmoy commited on
Commit
ccf7d04
1 Parent(s): 04e3b81

displays index updated date

Browse files
Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -3,6 +3,8 @@ import transformers
3
  import gradio as gr
4
  from ragatouille import RAGPretrainedModel
5
  from huggingface_hub import InferenceClient
 
 
6
 
7
  retrieve_results = 10
8
 
@@ -22,6 +24,17 @@ except:
22
  gr.Warning("Retriever not working!")
23
 
24
  mark_text = '# 🔍 Search Results\n'
 
 
 
 
 
 
 
 
 
 
 
25
 
26
  def rag_cleaner(inp):
27
  rank = inp['rank']
@@ -44,7 +57,7 @@ def get_rag(message):
44
  return get_references(message, RAG)
45
 
46
  with gr.Blocks(theme = gr.themes.Soft()) as demo:
47
- header = gr.Markdown("# ArXiv RAG")
48
  with gr.Group():
49
  msg = gr.Textbox(label = 'Search')
50
  with gr.Accordion("Advanced Settings", open=False):
 
3
  import gradio as gr
4
  from ragatouille import RAGPretrainedModel
5
  from huggingface_hub import InferenceClient
6
+ import re
7
+ from datetime import datetime
8
 
9
  retrieve_results = 10
10
 
 
24
  gr.Warning("Retriever not working!")
25
 
26
  mark_text = '# 🔍 Search Results\n'
27
+ header_text = "# ArXiv RAG\n"
28
+ try:
29
+ with open("README.md", "r") as f:
30
+ mdfile = f.read()
31
+ date_pattern = r'Index Last Updated : \d{4}-\d{2}-\d{2}'
32
+ match = re.search(date_pattern, mdfile)
33
+ date = match.group().split(': ')[1]
34
+ formatted_date = datetime.strptime(date, '%Y-%m-%d').strftime('%d %b %Y')
35
+ header_text += f'Index Last Updated: {formatted_date}\n'
36
+ except:
37
+ pass
38
 
39
  def rag_cleaner(inp):
40
  rank = inp['rank']
 
57
  return get_references(message, RAG)
58
 
59
  with gr.Blocks(theme = gr.themes.Soft()) as demo:
60
+ header = gr.Markdown(header_text)
61
  with gr.Group():
62
  msg = gr.Textbox(label = 'Search')
63
  with gr.Accordion("Advanced Settings", open=False):