Update app.py

app.py

@@ -11,7 +11,6 @@ from docx import Document
 from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
 from io import BytesIO
 
-# Set up OpenAI API
 openai.api_key = "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"
 
 st.markdown(
@@ -44,21 +43,6 @@ st.markdown(
         padding: 10px; /* Add padding to the headers */
         border-radius: 5px; /* Add border-radius to the headers */
     }
-    .feedback_button {
-        display: inline-block;
-        background-color: #e0e0e0;
-        padding: 10px;
-        margin: 5px;
-        border-radius: 10px;
-        box-shadow: 3px 3px 5px #bfbfbf, -3px -3px 5px #ffffff;
-        transition: box-shadow 0.3s ease;
-    }
-    .feedback_button:hover {
-        box-shadow: inset 3px 3px 5px #bfbfbf, inset -3px -3px 5px #ffffff;
-    }
-    .feedback_button:active {
-        box-shadow: inset 5px 5px 7px #bfbfbf, inset -5px -5px 7px #ffffff;
-    }
 </style>
 """,
     unsafe_allow_html=True,
@@ -71,23 +55,6 @@ model = CLIPModel().to(device)
 model.load_state_dict(torch.load("weights.pt", map_location=torch.device('cpu')))
 text_embeddings = torch.load('saved_text_embeddings.pt', map_location=device)
 
-def send_report_text(ctx, text):
-    server = Server.get_current()
-    if server:
-        for session in server._session_infos.values():
-            this_session = session.session
-            if this_session == ctx:
-                this_session.send_message({"type": "reportText", "text": text})
-
-def on_message(message, ctx):
-    if "type" in message and message["type"] == "getReport":
-        report_text = radiology_report_with_personal_info  # Replace this with the actual report text
-        send_report_text(ctx, report_text)
-
-def main():
-    ctx = st.session_state._get_report_ctx()
-    add_report_ctx(ctx, on_message)
-
 def download_link(content, filename, link_text):
     b64 = base64.b64encode(content).decode()
     href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">{link_text}</a>'
@@ -147,65 +114,15 @@ if uploaded_file is not None:
             # Add personal information to the radiology report
             radiology_report_with_personal_info = f"Patient Name: {first_name} {last_name}\nAge: {age}\nGender: {gender}\n\n{radiology_report}"
 
-            container = st.container()
-            with container:
-                st.header("Radiology Report")
-                st.write(radiology_report_with_personal_info)
-                st.markdown(download_link(save_as_docx(radiology_report_with_personal_info, "radiology_report.docx"), "radiology_report.docx", "Download Report as DOCX"), unsafe_allow_html=True)
-
-                ...
-                # Advanced Feedback System
-                st.header("Self Adaptive Learning: An Advanced AGI Feedback System")
-                st.markdown("RadiXGPT continuously learns and adapts to technical terms. Help us improve its performance by providing feedback::")
-                
-                # Load icons as base64
-                better_icon = "data:image/png;base64,iVBORw0KGg..."
-                satisfied_icon = "data:image/png;base64,iVBORw0KGg..."
-                worse_icon = "data:image/png;base64,iVBORw0KGg..."
-                
-                # Custom feedback buttons with neumorphic design and icons
-                feedback_html = f"""
-                <div style="display: flex; justify-content: center;">
-                    <div class="feedback_button" id="better_button" style="cursor:pointer;">
-                        <img src="{better_icon}" width="30" height="30" alt="Better Icon">
-                        <p>Better</p>
-                    </div>
-                    <div class="feedback_button" id="satisfied_button" style="cursor:pointer;">
-                        <img src="{satisfied_icon}" width="30" height="30" alt="Satisfied Icon">
-                <p>Satisfied</p>
-                    </div>
-                    <div class="feedback_button" id="worse_button" style="cursor:pointer;">
-                        <img src="{worse_icon}" width="30" height="30" alt="Worse Icon">
-                        <p>Regenerate Report...?</p>
-                    </div>
-                </div>
-                """
-                
-                st.markdown(feedback_html, unsafe_allow_html=True)
-                
-                # Define a function to handle feedback button clicks
-                def handle_feedback_button_click(feedback):
-                    if feedback == 'Regenerate Report...?':
-                        # Regenerate and display the radiology report
-                        radiology_report = generate_radiology_report(f"Write Complete Radiology Report for this: {caption}")
-                        radiology_report_with_personal_info = f"Patient Name: {first_name} {last_name}\nAge: {age}\nGender: {gender}\n\n{radiology_report}"
-                        container = st.container()
-                        with container:
-                            st.header("Radiology Report")
-                            st.write(radiology_report_with_personal_info)
-                            st.markdown(download_link(save_as_docx(radiology_report_with_personal_info, "radiology_report.docx"), "radiology_report.docx", "Download Report as DOCX"), unsafe_allow_html=True)
-                    # Implement other feedback handling logic here
-                
-                # Capture feedback button clicks
-                with st.empty():
-                    feedback_button_click = st.button("Better", key="better_button")
-                    if feedback_button_click:
-                        handle_feedback_button_click('better')
-                
-                    feedback_button_click = st.button("Satisfied", key="satisfied_button")
-                    if feedback_button_click:
-                        handle_feedback_button_click('satisfied')
-                
-                    feedback_button_click = st.button("Worse", key="worse_button")
-                    if feedback_button_click:
-                        handle_feedback_button_click('worse')
+            st.header("Radiology Report")
+            st.write(radiology_report_with_personal_info)
+            st.markdown(download_link(save_as_docx(radiology_report_with_personal_info, "radiology_report.docx"), "radiology_report.docx", "Download Report as DOCX"), unsafe_allow_html=True)
+
+            # Feedback buttons
+            st.header("Thanks for your feedback!")
+            feedback_options = ["Better", "Satisfied", "Worse"]
+            selected_feedback = st.radio("Please provide feedback on the generated report:", feedback_options)
+
+            if st.button("Submit Feedback"):
+                st.success("Thanks for providing feedback!")
+                # Implement your feedback handling logic here based on the `selected_feedback` value