Update app.py

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()