app.py

app.py

@@ -9,6 +9,7 @@ from docx import Document
 from datetime import datetime
 from pydub import AudioSegment
 from io import BytesIO
+import json
 
 # Streamlit Page Config
 st.set_page_config(page_title="Schlager ContractAi", layout="wide")
@@ -18,7 +19,7 @@ hide_huggingface_css = """
     <style>
         header {visibility: hidden;}  /* Hides the Hugging Face header */
         footer {visibility: hidden;}  /* Hides the footer */
-        .stDeployButton {display: none !important;}  /* Hides the "Stop" and deployment buttons */
+        .stDeployButton {display: none !important;}  /
         .stAppViewBlockContainer {padding-top: 0px !important;} /* Removes extra padding */
         iframe {display: none !important;} /* Hides the Hugging Face iframe */
     </style>
@@ -67,59 +68,15 @@ tab1, tab2, tab3, tab4 = st.tabs(["Contract", "Technical", "Minutes", "Document
 SUPPORTED_AUDIO_FORMATS = (".mp3", ".wav", ".m4a")
 SUPPORTED_TEXT_FORMATS = (".txt", ".docx", ".csv", ".xlsx", ".pdf")
 
-def install_dependencies():
-    try:
-        subprocess.run(['apt-get', 'update'], check=True)
-        subprocess.run(['apt-get', 'install', '-y', 'poppler-utils'], check=True)
-        subprocess.run(['apt-get', 'install', '-y', 'tesseract-ocr'], check=True)
-        subprocess.run(['apt-get', 'install', '-y', 'tesseract-ocr-eng'], check=True)
-        print("Packages installed successfully!")
-    except subprocess.CalledProcessError as e:
-        print(f"An error occurred: {e}")
-
-install_dependencies()
-
-def process_pdf(uploaded_file):
-    temp_pdf_path = "uploaded_document.pdf"
-    with open(temp_pdf_path, "wb") as f:
-        f.write(uploaded_file.read())
-    
-    os.system(f'pdftoppm -png "{temp_pdf_path}" img')
-    
-    extracted_text = ""
-    for image in os.listdir():
-        if image.startswith('img') and image.endswith('.png'):
-            output_txt = f"ocr_{image}.txt"
-            os.system(f'tesseract "{image}" "{output_txt[:-4]}"')
-            
-            with open(output_txt, "r") as txt_file:
-                extracted_text += txt_file.read() + "\n"
-    
-    for file in os.listdir():
-        if file.startswith('img') or file.startswith('ocr_img'):
-            os.remove(file)
-    os.remove(temp_pdf_path)
-    
-    return extracted_text
-
-with tab4:
-    st.subheader("Document Preparation")
-    uploaded_file = st.file_uploader("Upload a PDF (Max: 200MB)", type=["pdf"], accept_multiple_files=False)
-    
-    if uploaded_file:
-        st.write("Processing the uploaded document...")
-        extracted_text = process_pdf(uploaded_file)
-        
-        if extracted_text.strip():
-            st.text_area("Extracted Text", extracted_text, height=300)
-            st.download_button(
-                label="Download Extracted Text",
-                data=extracted_text,
-                file_name="extracted_text.txt",
-                mime="text/plain"
-            )
-        else:
-            st.error("No text could be extracted. Try another document.")
+def save_flagged_response(user_query, ai_response):
+    flagged_data = {
+        "query": user_query,
+        "response": ai_response,
+        "timestamp": datetime.now().isoformat()
+    }
+    with open("flagged_responses.json", "a") as file:
+        file.write(json.dumps(flagged_data) + "\n")
+    st.success("Response flagged for review.")
 
 # Contract Chat Section
 def contract_chat_section(tab, assistant_id, session_key, input_key):
@@ -137,7 +94,15 @@ def contract_chat_section(tab, assistant_id, session_key, input_key):
 
         for message in st.session_state[session_key]:
             role, content = message["role"], message["content"]
-            st.chat_message(role).write(content)
+            if role == "assistant":
+                col1, col2 = st.columns([5, 1])
+                with col1:
+                    st.chat_message(role).write(content)
+                with col2:
+                    if st.button("🚩 Flag", key=f"flag_{session_key}_{len(st.session_state[session_key])}"):
+                        save_flagged_response(st.session_state[session_key][-2]["content"], content)
+            else:
+                st.chat_message(role).write(content)
 
         if prompt := st.chat_input("Enter your message:", key=input_key):
             st.session_state[session_key].append({"role": "user", "content": prompt})