Spaces:
Running
Running
| # test_preloaded_model.py | |
| import gradio as gr | |
| import pickle | |
| import os | |
| # --- Dependencies needed for the Class Definition --- | |
| from transformers import pipeline | |
| import torch | |
| # --- Define the Class to Hold Both Pipelines --- | |
| # IMPORTANT: This exact class definition MUST be present here, | |
| # identical to the one in save_combined_model.py, for unpickling to work. | |
| class CombinedAnalyzer: | |
| """ | |
| A class to encapsulate sentiment analysis and AI text detection pipelines. | |
| NOTE: This definition must match the one used when saving the .pkl file. | |
| """ | |
| def __init__(self, sentiment_model_name="distilbert-base-uncased-finetuned-sst-2-english", | |
| detector_model_name="Hello-SimpleAI/chatgpt-detector-roberta"): | |
| print("Initializing CombinedAnalyzer structure...") | |
| self.device = 0 if torch.cuda.is_available() else -1 | |
| self.sentiment_model_name = sentiment_model_name | |
| self.detector_model_name = detector_model_name | |
| self.sentiment_pipeline = None | |
| self.detector_pipeline = None | |
| print(f"Class structure defined. Expecting pipelines for models: {sentiment_model_name}, {detector_model_name}") | |
| def analyze(self, text): | |
| """ | |
| Analyzes the input text for both sentiment and authenticity. | |
| """ | |
| if not isinstance(text, str) or not text.strip(): | |
| return "Error: Input text cannot be empty." | |
| results = [] | |
| # 1. Sentiment Analysis | |
| if self.sentiment_pipeline and callable(self.sentiment_pipeline): | |
| try: | |
| sentiment_result = self.sentiment_pipeline(text)[0] | |
| sentiment_label = sentiment_result['label'] | |
| sentiment_score = round(sentiment_result['score'] * 100, 2) | |
| results.append(f"Sentiment: {sentiment_label} (Confidence: {sentiment_score}%)") | |
| except Exception as e: | |
| results.append(f"Sentiment Analysis Error in loaded model: {e}") | |
| else: | |
| results.append("Sentiment Analysis: Model not available or not callable in loaded object.") | |
| # 2. AI Text Detection (Authenticity) | |
| if self.detector_pipeline and callable(self.detector_pipeline): | |
| try: | |
| detector_result = self.detector_pipeline(text)[0] | |
| auth_label_raw = detector_result['label'] | |
| auth_score = round(detector_result['score'] * 100, 2) | |
| if auth_label_raw.lower() in ['chatgpt', 'ai', 'generated']: | |
| auth_label_display = "Likely AI-Generated" | |
| elif auth_label_raw.lower() in ['human', 'real']: | |
| auth_label_display = "Likely Human-Written" | |
| else: | |
| auth_label_display = f"Label: {auth_label_raw}" | |
| results.append(f"Authenticity: {auth_label_display} (Confidence: {auth_score}%)") | |
| except Exception as e: | |
| results.append(f"AI Text Detection Error in loaded model: {e}") | |
| else: | |
| results.append("Authenticity: AI Text Detector model not available or not callable in loaded object.") | |
| return "\n".join(results) | |
| # --- Load the Model Automatically on Startup --- | |
| analyzer = None | |
| pickle_filename = "combined_analyzer.pkl" | |
| model_dir = "saved_model" | |
| pickle_filepath = os.path.join(model_dir, pickle_filename) | |
| model_load_error = None # Store potential loading error message | |
| print(f"Attempting to load pre-saved model from: {pickle_filepath}") | |
| try: | |
| print("\n--- SECURITY WARNING ---") | |
| print(f"Loading '{pickle_filepath}'. Unpickling data from untrusted sources is a security risk.") | |
| print("Ensure this .pkl file was created by you or a trusted source.\n") | |
| if not os.path.exists(pickle_filepath): | |
| raise FileNotFoundError(f"Model file not found at {pickle_filepath}") | |
| with open(pickle_filepath, 'rb') as f: | |
| analyzer = pickle.load(f) | |
| if not hasattr(analyzer, 'analyze') or not callable(analyzer.analyze): | |
| raise TypeError("Loaded object is not a valid analyzer (missing 'analyze' method).") | |
| else: | |
| print("Model loaded successfully.") | |
| sentiment_name = getattr(analyzer, 'sentiment_model_name', 'Unknown') | |
| detector_name = getattr(analyzer, 'detector_model_name', 'Unknown') | |
| print(f" -> Sentiment Model: {sentiment_name}") | |
| print(f" -> Detector Model: {detector_name}") | |
| except FileNotFoundError as e: | |
| model_load_error = f"ERROR loading model: {e}" | |
| print(model_load_error) | |
| print("Please ensure 'save_combined_model.py' was run successfully and") | |
| print(f"the file '{pickle_filename}' exists in the '{model_dir}' directory.") | |
| except (pickle.UnpicklingError, TypeError, AttributeError) as e: | |
| model_load_error = f"ERROR loading model: The pickle file might be corrupted, incompatible, or from a different version. Details: {e}" | |
| print(model_load_error) | |
| except Exception as e: | |
| model_load_error = f"An unexpected ERROR occurred during model loading: {e}" | |
| print(model_load_error) | |
| # --- Define the Gradio Analysis Function --- | |
| def analyze_text_interface(text_input): | |
| """Function called by Gradio to perform analysis using the pre-loaded model.""" | |
| if analyzer is None: | |
| # Use the stored error message if available | |
| error_msg = model_load_error or f"ERROR: The analyzer model could not be loaded from '{pickle_filepath}'." | |
| return error_msg | |
| if not text_input or not text_input.strip(): | |
| return "Please enter some text to analyze." | |
| print(f"Analyzing text: '{text_input[:60]}...'") | |
| try: | |
| results = analyzer.analyze(text_input) | |
| print("Analysis complete.") | |
| return results | |
| except Exception as e: | |
| print(f"Error during analysis: {e}") | |
| return f"An error occurred during analysis:\n{e}" | |
| # --- Build the Gradio Interface --- | |
| # **CORRECTION HERE:** Define the warning message string separately | |
| warning_message_text = "" | |
| if analyzer is None: | |
| # Use newline characters safely outside the f-string expression | |
| warning_message_text = "\n\n***WARNING: MODEL FAILED TO LOAD. ANALYSIS WILL NOT WORK.***" | |
| # Construct the full description string | |
| description_text = ( | |
| f"Enter text to analyze using the pre-loaded model from '{pickle_filepath}'.\n" | |
| "Checks for sentiment (Positive/Negative) and predicts if text is Human-Written or AI-Generated." | |
| f"{warning_message_text}" # Use the variable here | |
| ) | |
| interface = gr.Interface( | |
| fn=analyze_text_interface, | |
| inputs=gr.Textbox(lines=7, label="Text to Analyze", placeholder="Enter review text here..."), | |
| outputs=gr.Textbox(lines=7, label="Analysis Results", interactive=False), | |
| title="Sentiment & Authenticity Analyzer", | |
| description=description_text, # Use the constructed description string | |
| allow_flagging='never' | |
| ) | |
| # --- Launch the Interface --- | |
| if __name__ == "__main__": | |
| if analyzer is None: | |
| print("\n--- Interface launched, but MODEL IS NOT LOADED. Analysis will fail. ---") | |
| else: | |
| print("\n--- Launching Gradio Interface with pre-loaded model ---") | |
| interface.launch() |