Create app.py

app.py

@@ -0,0 +1,177 @@
+import gradio as gr
+from transformers import pipeline
+
+# Initialize the models
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+image_captioner = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
+sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
+
+# Function for text summarization
+def summarize_text(text):
+    if not text:
+        return "Please enter some text to summarize."
+    try:
+        result = summarizer(text, max_length=130, min_length=30, do_sample=False)
+        return result[0]['summary_text']
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Function for image captioning
+def caption_image(image):
+    if image is None:
+        return "Please upload an image."
+    try:
+        result = image_captioner(image)
+        return result[0]['generated_text']
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Function for sentiment analysis
+def analyze_sentiment(text):
+    if not text:
+        return "Please enter some text to analyze."
+    try:
+        result = sentiment_analyzer(text)
+        label = result[0]['label']
+        score = result[0]['score']
+        return f"Sentiment: {label}\nConfidence: {score:.2%}"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Create the Gradio interface with tabs
+with gr.Blocks(title="AI Wizard Toolkit") as demo:
+    gr.Markdown("""
+    # 🧙‍♂️ AI Wizard Toolkit
+    
+    Welcome to the AI Wizard Toolkit! This application provides three powerful AI tools:
+    
+    ## 🔮 Features:
+    
+    ### 📝 Tab 1: Text Summarizer
+    - **Model**: facebook/bart-large-cnn
+    - **Function**: Condense long articles or documents into concise summaries
+    - **How to use**: Paste your text in the input box and click "Summarize"
+    
+    ### 🖼️ Tab 2: Image Captioning
+    - **Model**: nlpconnect/vit-gpt2-image-captioning
+    - **Function**: Generate descriptive captions for your images
+    - **How to use**: Upload an image and click "Generate Caption"
+    
+    ### 😊 Tab 3: Sentiment Analysis
+    - **Model**: distilbert-base-uncased-finetuned-sst-2-english
+    - **Function**: Analyze the emotional tone of text (positive or negative)
+    - **How to use**: Enter your text and click "Analyze Sentiment"
+    
+    ---
+    
+    Select a tab below to get started!
+    """)
+    
+    with gr.Tabs():
+        # Tab 1: Text Summarization
+        with gr.Tab("📝 Text Summarizer"):
+            gr.Markdown("### AI-Powered Text Summarization")
+            gr.Markdown("Enter a long text and get a concise summary using facebook/bart-large-cnn model.")
+            
+            with gr.Row():
+                with gr.Column():
+                    text_input = gr.Textbox(
+                        label="Input Text",
+                        placeholder="Paste your article or long text here...",
+                        lines=10
+                    )
+                    summarize_btn = gr.Button("Summarize", variant="primary")
+                
+                with gr.Column():
+                    summary_output = gr.Textbox(
+                        label="Summary",
+                        lines=10
+                    )
+            
+            summarize_btn.click(
+                fn=summarize_text,
+                inputs=text_input,
+                outputs=summary_output
+            )
+            
+            gr.Examples(
+                examples=[
+                    ["The tower is 324 metres (1,063 ft) tall, about the same height as an 81-storey building, and the tallest structure in Paris. Its base is square, measuring 125 metres (410 ft) on each side. During its construction, the Eiffel Tower surpassed the Washington Monument to become the tallest man-made structure in the world, a title it held for 41 years until the Chrysler Building in New York City was finished in 1930. It was the first structure to reach a height of 300 metres. Due to the addition of a broadcasting aerial at the top of the tower in 1957, it is now taller than the Chrysler Building by 5.2 metres (17 ft)."]
+                ],
+                inputs=text_input
+            )
+        
+        # Tab 2: Image Captioning
+        with gr.Tab("🖼️ Image Captioning"):
+            gr.Markdown("### AI-Powered Image Captioning")
+            gr.Markdown("Upload an image and get an AI-generated caption using nlpconnect/vit-gpt2-image-captioning model.")
+            
+            with gr.Row():
+                with gr.Column():
+                    image_input = gr.Image(
+                        label="Upload Image",
+                        type="pil"
+                    )
+                    caption_btn = gr.Button("Generate Caption", variant="primary")
+                
+                with gr.Column():
+                    caption_output = gr.Textbox(
+                        label="Generated Caption",
+                        lines=3
+                    )
+            
+            caption_btn.click(
+                fn=caption_image,
+                inputs=image_input,
+                outputs=caption_output
+            )
+        
+        # Tab 3: Sentiment Analysis
+        with gr.Tab("😊 Sentiment Analysis"):
+            gr.Markdown("### AI-Powered Sentiment Analysis")
+            gr.Markdown("Analyze the sentiment of text using distilbert-base-uncased-finetuned-sst-2-english model.")
+            
+            with gr.Row():
+                with gr.Column():
+                    sentiment_input = gr.Textbox(
+                        label="Input Text",
+                        placeholder="Enter text to analyze sentiment...",
+                        lines=5
+                    )
+                    sentiment_btn = gr.Button("Analyze Sentiment", variant="primary")
+                
+                with gr.Column():
+                    sentiment_output = gr.Textbox(
+                        label="Sentiment Result",
+                        lines=5
+                    )
+            
+            sentiment_btn.click(
+                fn=analyze_sentiment,
+                inputs=sentiment_input,
+                outputs=sentiment_output
+            )
+            
+            gr.Examples(
+                examples=[
+                    ["I love this product! It's absolutely amazing and exceeded all my expectations."],
+                    ["This is terrible. I'm very disappointed and would not recommend it to anyone."],
+                    ["The weather is nice today."]
+                ],
+                inputs=sentiment_input
+            )
+    
+    gr.Markdown("""
+    ---
+    
+    ### 📖 About the Models:
+    
+    - **BART-large-CNN**: A transformer model fine-tuned for summarization tasks
+    - **ViT-GPT2**: Combines Vision Transformer with GPT-2 for image understanding
+    - **DistilBERT-SST-2**: A distilled version of BERT fine-tuned on sentiment classification
+    
+    All models are powered by Hugging Face Transformers! 🤗
+    """)
+
+if __name__ == "__main__":
+    demo.launch()