Update app.py

app.py

@@ -20,24 +20,29 @@ function(text_input, url_params) {
 """
 def predict(text, url_params):
     mod_url=""
+    mod=None
     try:
         mod_url = url_params.get('url')
         print (mod_url)
         out = gr.HTML.update(f"""<h1>Loading model from:<br>{mod_url} </h1>""")
     except Exception:
         out=gr.HTML.update(f"""<h1>Enter Model URL</h1>""")
-    return ["" + text + "", out,mod_url]
+    return ["" + text + "", mod,out,mod_url]
    
 def find_model(url):
-    try:
-        out = gr.HTML(f'''
-        <div>
-        <model-viewer src="{url}" camera-controls poster="https://cdn.glitch.global/fe6d4fc5-58dd-43de-a65d-7bd5b477e8e1/poster.webp?v=1696812691961" shadow-intensity="1"></model-viewer>
-        </div>
-        ''')
-    except Exception as e:
-        out = gr.HTML(f'''<h1>Enter Model URL</h1><br><h3>System Message: {e}''')
-    return out
+    mod=None
+    if url != "" and url != None:
+        try:
+            mod = gr.HTML(f'''
+            <div>
+            <model-viewer src="{url}" camera-controls poster="https://cdn.glitch.global/fe6d4fc5-58dd-43de-a65d-7bd5b477e8e1/poster.webp?v=1696812691961" shadow-intensity="1"></model-viewer>
+            </div>
+            ''')
+        except Exception as e:
+            out = gr.HTML(f'''<h1>Could not find Model</h1><br><h3>System Message: {e}''')
+    else:
+        out = gr.HTML(f'''<h1>Enter Model URL</h1>''')
+    return mod,out
 
     
 with gr.Blocks(css="style.css") as app:
@@ -45,8 +50,6 @@ with gr.Blocks(css="style.css") as app:
         with gr.Row():
             inp = gr.Textbox(label="URL")
             go_btn=gr.Button("Load")
-    
-            
         out_html=gr.HTML('''''')
     '''
     with gr.Tab("iFrame"):
@@ -56,7 +59,7 @@ with gr.Blocks(css="style.css") as app:
         </div>
         """)        
     '''
-    #markdown = gr.HTML("""""")
+    markdown = gr.HTML("""""")
 
     with gr.Row(visible=False):
         text_input=gr.Textbox()
@@ -65,7 +68,7 @@ with gr.Blocks(css="style.css") as app:
     
     
     app.load(None,None,None,_js=load_js1)
-    app.load(fn=predict, inputs=[text_input,url_params], outputs=[text_output,out_html,inp],_js=load_js2).then(find_model,inp,out_html)
+    app.load(fn=predict, inputs=[text_input,url_params], outputs=[text_output,out_html,markdown,inp],_js=load_js2).then(find_model,inp,out_html,markdown)
     go_btn.click(find_model,inp,out_html)
 app.launch()