Update app.py

app.py

@@ -1,18 +1,24 @@
+import streamlit as st
 from transformers import pipeline
 
-# Use a pipeline as a high-level helper
-pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
+# Title of the Streamlit app
+st.title("Email Reply Generator")
 
-# Function to generate a reply based on user's input email content
-def generate_reply(email_content):
+# Set up the pipeline for text generation using the replycraft model
+pipe = pipeline("text-generation", model="replycraft/replycraft", trust_remote_code=True)
+
+# Get email content from the user
+email_content = st.text_area("Enter the email content")
+
+if email_content:
+    # Prepare the messages and get the generated reply
     messages = [
         {"role": "user", "content": f"Generate a professional email reply to the following email: {email_content}"}
     ]
-    # Get the reply from the model
-    reply = pipe(messages)
-    return reply[0]['generated_text']
+    
+    # Generate the reply using the model
+    generated_reply = pipe(messages)
+    
+    # Display the generated reply
+    st.json(generated_reply)
 
-# Example usage
-user_email = "Dear team, I hope this email finds you well. Could you provide an update on the project status?"
-generated_reply = generate_reply(user_email)
-print(f"Generated Reply: {generated_reply}")