Update app.py

app.py

@@ -1,24 +1,44 @@
-import streamlit as st
 import os
+import subprocess
 import random
+import streamlit as st
 from groq import Groq
 import language_tool_python
 
+# Function to check and install Java
+def install_java():
+    try:
+        # Check if Java is installed
+        subprocess.run(["java", "-version"], check=True, capture_output=True)
+        st.write("Java is already installed.")
+    except FileNotFoundError:
+        # Install Java
+        st.write("Java is not installed. Installing now...")
+        subprocess.run(["apt-get", "update"], check=True)
+        subprocess.run(["apt-get", "install", "-y", "default-jre"], check=True)
+        st.write("Java installed successfully.")
+
+# Run the Java installation check
+install_java()
+
 # Initialize Groq client
 GROQ_API_KEY = "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"  # Replace with your Groq API key
 os.environ["GROQ_API_KEY"] = GROQ_API_KEY
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# Initialize Grammar Tool
+# Initialize LanguageTool
 try:
     tool = language_tool_python.LanguageTool('en-US')
-except ModuleNotFoundError as e:
-    tool = None
-    st.error("Java is not installed or not detected. Please install Java to enable grammar checking.")
+except Exception as e:
+    st.error("Error initializing LanguageTool: Please ensure Java is installed.")
+    raise e
 
 # Helper Functions
 def add_human_noise(output_text):
-    noise_phrases = ["you know,", "like I said,", "to be honest,", "frankly,", "truth be told,", "I mean,", "well, honestly,", "if I'm being real,"]
+    noise_phrases = [
+        "you know,", "like I said,", "to be honest,", "frankly,", 
+        "truth be told,", "I mean,", "well, honestly,", "if I'm being real,"
+    ]
     filler_words = ["uh,", "um,", "so,", "well,", "I guess,"]
     sentences = output_text.split(". ")
     noisy_sentences = [
@@ -34,6 +54,7 @@ def add_human_noise(output_text):
 def adjust_sentence_structure(output_text):
     sentences = output_text.split(". ")
     adjusted_sentences = []
+
     for sentence in sentences:
         words = sentence.split()
         if len(words) > 15:
@@ -42,15 +63,13 @@ def adjust_sentence_structure(output_text):
             adjusted_sentences.append(" ".join(words[midpoint:]))
         else:
             adjusted_sentences.append(sentence)
+    
     return ". ".join(adjusted_sentences)
 
 def check_grammar(output_text):
-    if tool:
-        matches = tool.check(output_text)
-        corrected_text = tool.correct(output_text)
-        return corrected_text
-    else:
-        return output_text
+    matches = tool.check(output_text)
+    corrected_text = tool.correct(output_text)
+    return corrected_text
 
 def refine_humanization(output_text, tone):
     refinement_prompt = (
@@ -70,7 +89,7 @@ def refine_humanization(output_text, tone):
         refined_text = check_grammar(refined_text)
         return refined_text
     except Exception:
-        return output_text
+        return output_text  # Return original output if refinement fails
 
 def split_text_into_chunks(text, max_words=500):
     words = text.split()
@@ -86,11 +105,23 @@ input_word_count = len(input_text.split()) if input_text.strip() else 0
 st.write(f"**Input Word Count:** {input_word_count}")
 
 # Options
-task_option = st.radio("Choose an option:", ("Humanize Text", "Rephrase Text"), index=0)
+task_option = st.radio(
+    "Choose an option:",
+    ("Humanize Text", "Rephrase Text"),
+    index=0
+)
 tone = st.selectbox("Select tone (for humanizing):", ["Casual", "Professional", "Neutral", "Engaging", "Friendly"])
 
-humanization_depth = st.slider("Select depth of humanization:", min_value=1, max_value=5, value=3, help="Higher values apply more refinements.")
+# Depth of Humanization
+humanization_depth = st.slider(
+    "Select depth of humanization:",
+    min_value=1,
+    max_value=5,
+    value=3,
+    help="Higher values apply more refinements to make the text appear less AI-like."
+)
 
+# Generate Output
 if st.button("Generate Output"):
     if not input_text.strip():
         st.error("Please enter some text to process.")
@@ -99,18 +130,40 @@ if st.button("Generate Output"):
             with st.spinner("Processing..."):
                 text_chunks = split_text_into_chunks(input_text, max_words=500)
                 output_chunks = []
+
                 for chunk in text_chunks:
-                    task_prompt = f"Make the following text highly human-readable in a {tone} tone: {chunk}" if task_option == "Humanize Text" else f"Rephrase the following text: {chunk}"
+                    if task_option == "Humanize Text":
+                        task_prompt = (
+                            f"Make the following text highly human-readable, engaging, and polished in a {tone} tone. "
+                            f"Focus on clarity, flow, and avoiding AI-detected phrasing: {chunk}"
+                        )
+                    else:
+                        task_prompt = f"Rephrase the following text while maintaining its original meaning: {chunk}"
+
                     chat_completion = client.chat.completions.create(
                         messages=[{"role": "user", "content": task_prompt}],
                         model="llama-3.3-70b-versatile",
                         stream=False,
                     )
+
                     output_text = chat_completion.choices[0].message.content.strip()
                     for _ in range(humanization_depth):
                         output_text = refine_humanization(output_text, tone)
                     output_chunks.append(output_text)
+
                 final_output = " ".join(output_chunks)
+                output_word_count = len(final_output.split())
+
+                st.success("Done!")
                 st.text_area("Generated Output:", value=final_output, height=300)
+                st.write(f"**Output Word Count:** {output_word_count}")
+
         except Exception as e:
-            st.error(f"An error occurred: {str(e)}")
+            st.error(f"An error occurred: {str(e)}")
+
+# Footer
+st.markdown("---")
+st.markdown(
+    "<p style='text-align: center; font-size: 14px;'>Designed by: <b>Engr. Makhdoom Muhammad Naeem Hashmi</b></p>",
+    unsafe_allow_html=True
+)