Update app.py

Please see if you can add an error message when user has not accepted the consent. As of now if the user has not given consent the button will not work.

app.py

@@ -2,10 +2,13 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 # Initialize the Hugging Face Inference Client
-client = InferenceClient()
+client = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct")
 
 # Function to stream the compliance suggestions as they are generated
-def analyze_compliance_stream(code, compliance_standard):
+def analyze_compliance_stream(code, compliance_standard, consent):
+    if not consent:
+        return "You must consent to the processing of your code snippet."
+
     prompt = f"Analyze the following code for {compliance_standard} compliance and suggest modifications or refactoring to meet the guidelines:\n\n{code}"
     
     messages = [
@@ -14,10 +17,9 @@ def analyze_compliance_stream(code, compliance_standard):
 
     # Create a stream to receive generated content
     stream = client.chat.completions.create(
-        model="Qwen/Qwen2.5-Coder-32B-Instruct",
         messages=messages,
         temperature=0.5,
-        max_tokens=1024,
+        max_tokens=32000,
         top_p=0.7,
         stream=True
     )
@@ -28,10 +30,18 @@ def analyze_compliance_stream(code, compliance_standard):
         compliance_suggestions += chunk.choices[0].delta.content
         yield compliance_suggestions  # Yield incremental content to display immediately
 
+    # Delete the code snippet after processing
+    del code
+
 # Create Gradio interface with the modified layout
 with gr.Blocks() as app:
     gr.Markdown("## Code Compliance Advisor")
     gr.Markdown("Analyze your code for legal compliance and security standards (e.g., GDPR, HIPAA) and receive actionable suggestions.")
+    gr.Markdown("### Privacy Notice")
+    gr.Markdown("""
+    By using this tool, you consent to the processing of your code snippet for compliance analysis. Your code will be sent to a third-party model for analysis and the results will be displayed to you. We do not store your code or the results unless required by law. 
+    You have the right to access and erase your data. If you wish to request access to your data or have it erased, please contact us at [[email protected]].
+    """)
 
     with gr.Row():
         # First column for input components
@@ -42,6 +52,7 @@ with gr.Blocks() as app:
                 label="Compliance Standard",
                 value="GDPR"
             )
+            consent_checkbox = gr.Checkbox(label="I consent to the processing of my code snippet", value=False)
             analyze_button = gr.Button("Analyze Compliance")
 
         # Second column for output
@@ -50,7 +61,7 @@ with gr.Blocks() as app:
             output_markdown = gr.Markdown()
 
     # Link button to function with inputs and outputs
-    analyze_button.click(fn=analyze_compliance_stream, inputs=[code_input, compliance_standard], outputs=output_markdown)
+    analyze_button.click(fn=analyze_compliance_stream, inputs=[code_input, compliance_standard, consent_checkbox], outputs=output_markdown)
 
 # Run the Gradio app
 app.launch()