updated app.py

app.py

@@ -1,28 +1,34 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load model with custom generation settings
-pipe = pipeline(
+# Load both summarization and sentiment models
+summarizer = pipeline(
     "summarization",
     model="Manish014/review-summariser-gpt-config1",
     tokenizer="Manish014/review-summariser-gpt-config1",
-    device=0  # Use GPU if available, else set to -1
+    device=0  # Use GPU if available
 )
 
-# Summarization function with settings
-def summarize(text):
+sentiment_analyzer = pipeline("sentiment-analysis")
+
+# Inference function
+def analyze_review(text):
     if not text.strip():
-        return "Please enter a product review."
-    
-    result = pipe(
+        return "Please enter a review.", "Sentiment unavailable."
+
+    summary = summarizer(
         text,
         max_length=80,
         min_length=10,
         num_beams=4,
         early_stopping=True,
         length_penalty=1.2
-    )
-    return result[0]["summary_text"]
+    )[0]["summary_text"]
+
+    sentiment = sentiment_analyzer(text)[0]
+    sentiment_output = f"{sentiment['label']} ({round(sentiment['score'] * 100, 2)}%)"
+
+    return summary, sentiment_output
 
 # Example reviews
 example_reviews = [
@@ -31,17 +37,30 @@ example_reviews = [
     ["Worst purchase I've made. Do not recommend at all."]
 ]
 
-# Gradio Interface
-demo = gr.Interface(
-    fn=summarize,
-    inputs=gr.Textbox(label="Product Review", lines=5, placeholder="Enter product review..."),
-    outputs=gr.Textbox(label="Summary"),
-    title="📝 Review Summariser GPT - Config 1",
-    description="Enter a long product review and get a helpful short summary.",
-    examples=example_reviews,
-    theme="default"
-)
+# Build Gradio UI
+with gr.Blocks(theme=gr.themes.Base()) as demo:
+    gr.Markdown("## 📝 Review Summariser GPT - Config 1")
+    gr.Markdown("Enter a long product review below to get a helpful short summary and sentiment prediction.")
+
+    with gr.Row():
+        input_box = gr.Textbox(label="🗣️ Product Review", lines=5, placeholder="Write your detailed review here...")
+
+    with gr.Row():
+        summary_output = gr.Textbox(label="✂️ Summary")
+        sentiment_output = gr.Textbox(label="📊 Sentiment")
+
+    submit_btn = gr.Button("🔍 Analyze Review")
+    clear_btn = gr.Button("🧹 Clear")
+
+    submit_btn.click(analyze_review, inputs=input_box, outputs=[summary_output, sentiment_output])
+    clear_btn.click(lambda: ("", "", ""), outputs=[input_box, summary_output, sentiment_output])
+
+    gr.Examples(
+        examples=example_reviews,
+        inputs=input_box,
+        label="🔁 Try an Example"
+    )
 
-# Launch app
+# Run the app
 if __name__ == "__main__":
     demo.launch()