Update app.py

app.py

@@ -1,179 +1,11 @@
-import os
-import gradio as gr
-from smolagents import (
-    tool,
-    CodeAgent,
-    DuckDuckGoSearchTool,
-    InferenceClientModel,
-    FinalAnswerTool,
-)
-from huggingface_hub import InferenceClient
-from fastapi import FastAPI
-import tempfile
-from PIL import Image
-
-def pil_to_tempfile(image):
-   
-    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
-    tmp_path = tmp.name
-    tmp.close()
-
-    image.save(tmp_path, format="PNG")
-
-    return tmp_path
-
-token = os.getenv("HF_TOKEN")
-
-client = InferenceClient(token=token)
-
-nsfw_image_detection_client = InferenceClient(
-    provider="hf-inference",
-    api_key=token
-)
-
-text_to_image_client = InferenceClient(
-    model="stabilityai/stable-diffusion-3-medium",
-    api_key=token
-)
-
-@tool
-def nsfw_detection_tool(nsfw_detection_input:  Image.Image) -> str:
-    """
-    Suitable for filtering through score explicit or inappropriate content in images.
-    Args:
-        nsfw_detection_input (Image.Image): The image to check.
-    Returns:
-        str: Highest score result.
-    """ 
-    try:
-        outputs = client.image_classification(
-            nsfw_detection_input,
-            model="Falconsai/nsfw_image_detection"
-        )
-        top_result = max(outputs, key=lambda x: x.score)
-        return f"{top_result.label.upper()}: {top_result.score:.1%}"
-    except Exception as e:
-        return f"Detection failed: {str(e)}"
-
-image_output = None
-
-@tool
-def image_tool(prompt: str) -> str:
-    """
-    Generate an image from text using SD3-Medium.
-    Args:
-        prompt (str): image description
-    Returns:
-        str: A confirmation message.
-    """    
-    global image_output
-    
-    try:
-        image = text_to_image_client.text_to_image(
-            prompt=prompt,
-            negative_prompt="blurry, distorted, low quality",
-            guidance_scale=7.0,
-            num_inference_steps=28,
-            width=1024,
-            height=1024
-        )
-        image_output = image
-        return "Image successfully generated and stored for Gradio UI."
-        
-    except Exception as e:
-        image_output = None
-        print(f"Image generation failed: {e}")
-        return f"Image generation failed: {e}"
-   
-@tool
-def search_tool(query: str)-> str:
-    """
-    Search the web and return the most relevant results.
-
-    Args:
-        query (str): The search query.
-
-    Returns:
-        str: The search results.
-    """
-    web_search_tool = DuckDuckGoSearchTool(max_results=5, rate_limit=2.0)
-    
-    results = web_search_tool(query)
-    
-    return results
-
-@tool
-def sentiment_tool(text: str) -> str:
-    """
-    Analyze sentiment of given text.
-
-    Args:
-        text (str): The sentiment query.
-    
-    Returns: str: sentiment
-    """
-    messages = [
-        {"role": "system", "content": "Analyze the sentiment of the following text using a range score of 0 -> 10 and provied alternative wording"},
-        {"role": "user", "content": text},
-    ]   
-    
-    completion = client.chat.completions.create(
-        model="meta-llama/Llama-3.3-70B-Instruct",
-        messages=messages,
-        max_tokens=150,
-    )
-
-    result = completion.choices[0].message.content
-
-    return result
-
-final_answer = FinalAnswerTool()
-
-model = InferenceClientModel(
-    model_id="meta-llama/Llama-3.3-70B-Instruct",
-    token=token,
-    max_tokens=2096,
-    temperature=0.5,
-)
-
-agent = CodeAgent(
-    model=model,
-    tools=[
-        image_tool,
-        nsfw_detection_tool,
-        sentiment_tool,
-        search_tool,
-        final_answer,
-    ],
-    max_steps=6,
-    planning_interval=None,
-)
-
-agent.prompt_templates["system_prompt"] += """
-    You are a tool calling agent.
-    You have access to these tools: 
-    - sentiment_tool(text: str) -> str 
-    - Analyze sentiment of given text.
-    - search_tool(query: str) -> str
-    - Search the web and return the most relevant results.
-    - Used for sentiment analysis
-    - image_tool(prompt: str) -> str
-    - Generate an image from a text prompt, if successfull or not you will be notified by the return string.
-    - nsfw_detection_tool(nsfw_detection_input: Image.Image) -> str
-    - The nsfw_detection_input additional argument is processed entirely within the tool to produce a score from the input.
-    - You must construct a well-formatted human-readable answer
-    - You must introduce yourself as Jerry and greet the user in the answer
-    - You must try include newlines, bullets, numbering, and proper punctuation
-    - You must use this answer in final_answer
-"""
-
-def run_agent(query: str, nsfw_detection_input: Image.Image):
+def run_agent(query, nsfw_detection_input):
+    # Your existing run_agent function here
     global image_output
     image_output = None
-
+    
     try:
         response = agent.run(
-            query, 
+            query if query else "", 
             additional_args={"nsfw_detection_input": nsfw_detection_input}
         )
         return image_output, str(response)
@@ -183,25 +15,31 @@ def run_agent(query: str, nsfw_detection_input: Image.Image):
 with gr.Blocks(title="Jerry AI Assistant") as demo:
     gr.Markdown("# 🤖 Jerry - Your AI Assistant")
     
+    # Shared output components
+    agent_response = gr.Textbox(
+        label="Response",
+        lines=5,
+        interactive=False
+    )
+    
     with gr.Tab("Chat"):
         with gr.Row():
-            query = gr.Textbox(
+            query_chat = gr.Textbox(
                 lines=3, 
                 label="Ask me anything...",
-                placeholder="Generate an image of a cat, analyze its sentiment, etc."
+                placeholder="Generate an image of a cat, analyze its sentiment, etc.",
+                scale=4
             )
         
         with gr.Row():
-            run_btn = gr.Button("🚀 Run", variant="primary", scale=1)
-            clear_btn = gr.Button("🗑️ Clear", scale=0)
+            run_chat_btn = gr.Button("🚀 Run", variant="primary", scale=1)
+            clear_chat_btn = gr.Button("🗑️ Clear", scale=0)
         
-        agent_response = gr.Textbox(
-            label="Response",
-            lines=5,
-            interactive=False
+        run_chat_btn.click(
+            fn=run_agent, 
+            inputs=[query_chat, gr.Image(visible=False)],  # Hidden image input for chat
+            outputs=[gr.Image(visible=False), agent_response]
         )
-
-        run_btn.click(fn=run_agent, inputs=[query, None], outputs=[None, agent_response])
     
     with gr.Tab("Image Tools"):
         with gr.Row():
@@ -214,8 +52,29 @@ with gr.Blocks(title="Jerry AI Assistant") as demo:
                 label="Generated Image",
                 height=300
             )
-
-            run_btn.click(fn=run_agent, inputs=[None, nsfw_detection_input], outputs=[image_output, agent_response])
+        
+        with gr.Row():
+            query_img = gr.Textbox(
+                lines=2,
+                label="Image generation prompt",
+                placeholder="A beautiful sunset over mountains..."
+            )
+        
+        with gr.Row():
+            run_img_btn = gr.Button("🎨 Generate Image", variant="primary")
+            check_nsfw_btn = gr.Button("🔍 Check NSFW")
+        
+        run_img_btn.click(
+            fn=run_agent,
+            inputs=[query_img, gr.Image(visible=False)],
+            outputs=[image_output, agent_response]
+        )
+        
+        check_nsfw_btn.click(
+            fn=run_agent,
+            inputs=[gr.Textbox(visible=False), nsfw_detection_input],
+            outputs=[gr.Image(visible=False), agent_response]
+        )
     
     gr.Examples(
         examples=[
@@ -223,8 +82,6 @@ with gr.Blocks(title="Jerry AI Assistant") as demo:
             "What's the sentiment of: I love this product!",
             "Search for latest AI news"
         ],
-        inputs=query,
+        inputs=query_chat,
         label="Quick Examples"
     )
-
-demo.launch(server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft())