Update app.py

app.py

@@ -393,23 +393,49 @@ def find_rss():
     yield out_box,[(None,f'Source is current as of:\n{timestamp} UTC\n\nThe current Date and Time is:\n{timestamp} UTC')]
 
     
-def load_data():
-    yield None,[(None,f'Loading data source, please wait')]
-    
-    f_ist = (api.list_repo_files(repo_id=reponame, repo_type="dataset"))
-    f_ist.sort(reverse=True)
-    print(f_ist)
-    r = requests.get(f'{save_data}{f_ist[0]}') 
-    lod = json.loads(r.text)
+def load_data(inp=None):
     timestamp=str(datetime.datetime.now())
-    filename=f_ist[0].split("/")[1].split(".json")[0].replace("--"," ")
-    print (filename)
-    filename_start = filename.split(" ")[0]
-    filename_end = filename.split(" ")[1]
-    filename_end = filename_end.replace("-"[0],":").replace("-"[0],":").replace("-"[0],".")
-    #filename_end_far=filename_end.split(":")[2]
-    print (filename)
-    yield lod,[(None,f'Source is current as of:\n{filename_start} {filename_end} UTC\n\nThe current Date and Time is:\n{timestamp} UTC')]
+    yield None,[(None,f'Loading data source, please wait')]
+    if inp:
+        yield None,[(None,f'Loading data from {inp}, please wait')]
+        
+        r = requests.get(f'{inp}') 
+        if r.status_code == 200:
+            lod={}
+            try:
+                if ".json" in rss_url:
+                    lod = json.loads(r.text)
+                if ".xml" in rss_url:
+                    lod = xmltodict.parse(r.content)
+                if ".rss" in rss_url:
+                    lod = xmltodict.parse(r.content)
+                else:
+                    try:
+                        lod = xmltodict.parse(r.content)
+                    except Exception as e:
+                        yield None, [(None, f'{rss_url} ::ERROR:: {e}')]
+            except Exception as e:
+                yield None, [(None, f'{rss_url} ::ERROR:: {e}')]
+            yield lod,[(None,f'Source is current as of:\n{timestamp} UTC')]
+        else: 
+            lod =
+            yield None, [(None, f'{rss_url} ::ERROR::COULD NOT CONNECT:: {r.status_code}')]
+    if not inp:
+        yield None,[(None,f'Loading data from database, please wait')]
+        
+        f_ist = (api.list_repo_files(repo_id=reponame, repo_type="dataset"))
+        f_ist.sort(reverse=True)
+        print(f_ist)
+        r = requests.get(f'{save_data}{f_ist[0]}') 
+        lod = json.loads(r.text)
+        filename=f_ist[0].split("/")[1].split(".json")[0].replace("--"," ")
+        print (filename)
+        filename_start = filename.split(" ")[0]
+        filename_end = filename.split(" ")[1]
+        filename_end = filename_end.replace("-"[0],":").replace("-"[0],":").replace("-"[0],".")
+        #filename_end_far=filename_end.split(":")[2]
+        print (filename)
+        yield lod,[(None,f'Source is current as of:\n{filename_start} {filename_end} UTC\n\nThe current Date and Time is:\n{timestamp} UTC')]
     
 with gr.Blocks() as app:
     cb = gr.Chatbot(height=600, show_share_button=True, show_copy_button=True)
@@ -417,14 +443,16 @@ with gr.Blocks() as app:
         inst = gr.Textbox(label="Instructions")
         sub_btn=gr.Button("Submit")
     with gr.Row():
+        rss_custom=gr.Textbox("URL for RSS feed (.rss,.xml)")
         load_btn = gr.Button("Load RSS")
+    with gr.Accordion():
         u_btn=gr.Button("Update [RSS Data]")
         keyw = gr.Button("Use Keyword [Experimental]")
     with gr.Row():
         out_json = gr.JSON()
         fil = gr.Textbox()
     keyw.click(get_records,[inst,out_json],[inst,cb])
-    load_btn.click(load_data,None,[out_json,cb])
+    load_btn.click(load_data,rss_custom,[out_json,cb])
     u_btn.click(find_rss,None,[out_json,cb])
     sub_btn.click(summarize,[inst,cb,out_json],[inst,cb])
 app.queue(default_concurrency_limit=20).launch()