Update app.py

app.py

@@ -2,7 +2,36 @@ import os
 import spaces
 from huggingface_hub import login
 import gradio as gr
-from smolagents import HfApiModel, CodeAgent, load_tool
+from smolagents import HfApiModel, CodeAgent
+
+
+
+#  Login to Hugging Face for access to LLama
+login(token = os.getenv('hf_login'))
+
+
+
+execute_firecrawl = firecrawler.execute_firecrawl
+agent = CodeAgent(
+    tools=[execute_firecrawl], model=HfApiModel("meta-llama/Llama-3.3-70B-Instruct"), max_steps=4
+)
+
+
+def get_answer(question):
+    """
+    A function that takes any question as input and returns the answer using agent.run()
+    
+    Args:
+        question (str): The question to be answered
+        
+    Returns:
+        str: The answer to the question
+    """
+    answer = agent.run(question)
+    answer = answer
+    print("Final output:")
+    print(answer)
+    return answer
 
 
 
@@ -21,7 +50,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     
     with gr.Row():
         with gr.Column(scale=1, min_width=300):
-            input_image = gr.Image(label="Upload a suspicious screenshot (email, text message, advertisement etc)", type="pil")
+            input_image =  gr.Image(label="Upload a suspicious screenshot (email, text message, advertisement etc)", type="pil")
             input_text = gr.Textbox(label="Description", info="Please describe your concerns regarditng the situation",lines=3,value="On Monday the 5th of April...")
             input_url = gr.Textbox(label="URLs", info="Please enter any suspicious URLs assosiated with the situation",lines=3,value="www.biz.xyz, www.freemasons.io, ...")
             btn = gr.Button("Process submission")
@@ -29,13 +58,11 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
 
         
         with gr.Column(scale=2, min_width=300):
-            #t1 = gr.Textbox(label='OCR text')
-            #t2 = gr.Textbox(label='Description')
             t3 = gr.Textbox(label="Advice")
             
     # Output button 
     btn.click(
-        fn=describe_image, 
+        fn=get_answer, 
         inputs=[input_image, input_text, input_url], 
         outputs=[t3]
     )